search

prateekiiest / titanic_survival_exploration

Udacity Machine Learning Nano degree Program Project Predicting Passenger Survival
MIT License
21 stars 35 forks source link

Building various Classification Models and comparing their accuracy #25

Closed nilaypathak closed 6 years ago

nilaypathak commented 6 years ago

I have used confusion matrix and split the dataset into train and test sets in order to compare the relative accuracy of 4 different classification models I have used the top six relevant features in making the models namely

  1. pclass
  2. sex
  3. Age
  4. sibsp
  5. parch
  6. fare The graph is based on the results of the confusion matrix.