Skip to content

jademene/Titanic-Survival-Classification-Model

 
 

Repository files navigation

Titanic survival prediction model

Using the 'train.csv' from the Titanic - Machine Learning from Disaster Kaggle competition, a model that predicts which passengers survived the Titanic shipwreck will be created.

This will be done using PySpark on Google Colab.

What will you find in this repo:

  • Data cleaning and EDA
  • Classifications algorithms considered
  • Hyperparameter tuning
  • Compare the best models
  • Predict survival of Titanic passengers

Note: this was a collaborative task with three other peers in the inernational BABD master @ POLIMI General School of Management.

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 100.0%