In this challenge, we will try to predict if a passenger survived or not during the Titanic disaster. The dataset is available on Kaggle. We use three different models : Logistic Regression from the scikit-learn library, XGBoost and LightGBM.
LogReg.py
: Logistic Regression model
XGBoost.py
: XGBoost model
LightGBM.py
: LightGBM model
Model | Accuracy |
---|---|
LightGBM | 0.77511 |
XGBoost | 0.77511 |
Logistic Regression | 0.76794 |