My first machine learning project! Predicting passenger survival on the Titanic using Python.
- Goal: Predict whether a Titanic passenger survived or not
- Dataset: Kaggle Titanic dataset (891 passengers)
- Accuracy Achieved: 81.01%
- Python 3.9
- pandas (data manipulation)
- matplotlib (visualization)
- scikit-learn (machine learning)
- Females had 74.2% survival rate vs males 18.9%
- 1st class passengers had 63% survival vs 3rd class 24.2%
- Age, fare, and family size also influenced survival
- Data exploration and visualization
- Handling missing values (Age, Embarked)
- Feature engineering (converting text to numbers)
- Train/test split (80/20)
- Logistic Regression model training
- Model evaluation
titanic.py- Main Python scripttrain.csv- Training datasetREADME.md- This file
pip install pandas numpy matplotlib scikit-learn
python3 titanic.py- Complete machine learning workflow
- Data cleaning and preparation
- Model training and evaluation
- The importance of exploratory data analysis
- Try Random Forest or Decision Trees
- Feature engineering (family size, title extraction)
- Hyperparameter tuning
- Ensemble methods
This is my first data science project as I learn Python and machine learning!
Dataset from Kaggle Titanic Competition