Titanic Survival Classifier
A Random Forest classifier trained on the classic Titanic dataset to predict passenger survival.
Model Details
- Model type: Random Forest (scikit-learn)
- Trees: 200, max_depth=6
- Features: 12 engineered features (Pclass, Sex, Age, SibSp, Parch, Fare, Embarked, Title, FamilySize, IsAlone, FarePerPerson, HasCabin)
- Dataset: phihung/titanic (891 samples)
Performance
| Metric | Score |
|---|---|
| Test Accuracy | 80.0% |
| Test F1 (survived) | 0.72 |
| CV Accuracy (5-fold) | 82.5% |
Usage
Generated by ML Intern
This model repository was generated by ML Intern, an agent for machine learning research and development on the Hugging Face Hub.
- Try ML Intern: https://smolagents-ml-intern.hf.space
- Source code: https://github.com/huggingface/ml-intern
- Downloads last month
- -