“Haberman's Survival Data" Set

Alyssa
Alyssa New Altair Community Member
edited November 2024 in Community Q&A
I am looking to find the best performing classification model to predict if a patient will survive 5 years or longer after the surgery. I need to do a comparison for my class. Which, if any or all, of the below Classification models should I use?:

·         Decision tree

·         K nearest

·         Naïve bayes

·         ANN

·         SVM

·         Random forest

Thanks so much!

Tagged:

Answers

  • jacobcybulski
    jacobcybulski New Altair Community Member
    All these models are applicable to the classification task, so I suggest you try them all and see which gives the best validation performance. In the process you may need to explore your data and see if any pre-processing is required for each particular model and if distribution or label values indicates which performance measures are applicable to your data. Good luck.