Caihanwang / Stroke_Prediction

0 stars 0 forks source link

model selection #2

Open Caihanwang opened 3 years ago

Caihanwang commented 3 years ago

We are going to choose a model from logistic regression, SVM, Decision tree, Random forest and XGBoost. We do the model diagnose with statistical methods to find the most accurate one to adjust for our data. The model packages could be imported from sklearn.

patrickkwang commented 3 years ago

What is the deliverable for this project? A Python library? Command-line application?

Caihanwang commented 3 years ago

We decided to create a command-line application with a user interface.