ikmckenz / target-pred-py

A simple machine learning model for small-molecule target prediction in Python.
GNU General Public License v3.0
18 stars 8 forks source link

Add "top five" accuracy #11

Closed ikmckenz closed 5 years ago

ikmckenz commented 5 years ago

Right now when training the model we use the built-in classification_report and confusion_matrix scikit-learn to judge model accuracy. We should add a "top five" accuracy measure to this, as when predicting we return the top 5 predicted classes by default. Top Five accuracy would show the percentage of test predictions which have the correctly predicted label in the top five returned classes.