Skip to content

yutongxie58/titanic-machine-learning-from-disaster

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 

Repository files navigation

Titanic: Machine Learning from Disaster

Project Description

This repository contains my solution for the Kaggle competition, "Titanic: Machine Learning from Disaster." The challenge is to predict the survival of passengers aboard the Titanic based on various features such as age, gender, ticket class, and more.

Dataset Information

The dataset consists of two main files:

  • train.csv: This file contains the training data, including features and the ground truth (survival status) for a subset of passengers.
  • test.csv: This file contains the test data, where the goal is to predict the survival status of the passengers.

For more details about the dataset and competition, please refer to the Kaggle Titanic competition page.

Approach

For this project, I used a Random Forest model to predict passenger survival. Random Forest is a versatile and powerful machine learning algorithm that often performs well on a variety of datasets.

I followed these steps in my approach:

  1. Data Preprocessing: Handling missing values, encoding categorical features, and feature scaling.
  2. Model Building: Creating a Random Forest classifier.
  3. Model Training: Fitting the model on the training data.
  4. Model Evaluation: Assessing the model's performance using appropriate evaluation metrics.
  5. Prediction: Making predictions on the test data.
  6. Submission: Preparing and submitting the predictions on Kaggle.

The code and detailed analysis can be found in the Jupyter Notebook provided in this repository.

Results

My Random Forest model achieved an accuracy of 87% on the test data.

Acknowledgments

I'd like to thank Kaggle for providing this exciting competition and the data science community for their valuable insights and contributions.

Feel free to explore the code and analysis in the Jupyter Notebook and provide any feedback or suggestions.

About

A titanic survival prediction using Random Forest.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published