Commit 8aefbea9 by Pamela Osuna

micro-average for the roc_auc

parent 01e0840d
......@@ -13,7 +13,7 @@ def roc_auc(n_classes, y_test, y_score):
fpr["micro"], tpr["micro"], _ = roc_curve(y_test.ravel(), y_score.ravel())
roc_auc["micro"] = auc(fpr["micro"], tpr["micro"])
return fpr[2], tpr[2], roc_auc[2]
return fpr["micro"], tpr["micro"], roc_auc["micro"]
def plot_roc_auc(fpr, tpr, roc_auc):
plt.figure()
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment