teejlab / API-Risk-Assessment-Framework

A framework for quantifying API risks.
https://teejlab.github.io/API-Risk-Assessment-Framework/intro.html
MIT License
5 stars 8 forks source link

Model selection for prediction pipeline #56

Open Anupriya-Sri opened 2 years ago

Anupriya-Sri commented 2 years ago

We have trained, tuned and evaluated the following models:

While tree based models are giving a slightly better performance than Logistic Regression, I suggest we use Logistic Regression because it is easy to interpret and would allow the partner to use it to understand feature importances in future.

This issue has been opened to discussion about the team's perspectives on model selection.