Would you Survive Titanic? 🌊

Here is the streamlit app deployed in Heroku!


The home screen of the webapp.

Bring out the time machine and travel back to 1912! Go on Titanic's maiden voyage and see if you would survive its titanic crash.

This app uses a Decision Tree trained on the Titanic Dataset to predict whether you will survive or not.

The machine learning model was trained in this Kaggle Notebook.

The aim of this project is to make a webapp using Streamlit and deploy it in Heroku.

This is why only four features are chosen for the machine learning model to learn:

Pclass (Ticket Class) : This variable denotes the class of the ticket for Titanic ship. The options are 1st Class, 2nd Class and 3rd Class.
Sex : This indicates the sex of the person boarding the ship. The options are male and female.
Age : This variable is the age of the person in years. Only integer values are accepted and this ranges from 0 to 100.
Is_alone : This variable denotes whether the person is travelling alone or not. The options are simply Yes and No.

The features were carefully chosen so that the user does not have any security concerns. To my best knowledge, no data is collected via this app.

Huge shout out to Data Professor for his video on how to deploy machine learning models to Heroku.

I would appreciate any feedback regarding this project.

Do check out the app and have fun with it!

p.s. Not liable if you drown in Titanic despite the model's prediction. 😅

Some Screenshots from the app


Predictions by the model on the chances of survival for Jack and Rose.


Taking Inputs for the model to make predictions on.