junos
/
rapids
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Minor fix of metrics section

pull/98/head
Meng Li 2020-08-17 17:27:58 -04:00
parent 6eaf0f5724
commit b3fbc79195
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/models/modeling_utils.py
View File

@ -49,7 +49,10 @@ def getMetrics(pred_y, pred_y_prob, true_y):
metrics = {} metrics = {}
# metrics for all categories # metrics for all categories
metrics["accuracy"] = accuracy_score(true_y, pred_y) metrics["accuracy"] = accuracy_score(true_y, pred_y)
metrics["auc"] = roc_auc_score(true_y, pred_y_prob) try:
metrics["auc"] = roc_auc_score(true_y, pred_y_prob)
except:
metrics["auc"] = None
metrics["kappa"] = cohen_kappa_score(true_y, pred_y) metrics["kappa"] = cohen_kappa_score(true_y, pred_y)
# metrics for label 0 # metrics for label 0
metrics["precision0"] = precision_score(true_y, pred_y, average=None, labels=[0,1], zero_division=0)[0] metrics["precision0"] = precision_score(true_y, pred_y, average=None, labels=[0,1], zero_division=0)[0]