The Altair Community is migrating to a new platform to provide a better experience for you. In preparation for the migration, the Altair Community is on read-only mode from October 28 - November 6, 2024. Technical support via cases will continue to work as is. For any urgent requests from Students/Faculty members, please submit the form linked here

“Haberman's Survival Data" Set

AlyssaAlyssa Member Posts: 1 Learner I
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!

Answers

  • jacobcybulskijacobcybulski Member, University Professor Posts: 391 Unicorn
    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.
Sign In or Register to comment.