Commit 3d2c33d6 by Pamela Osuna

uses the renamed function create_pr

parent ed21820f
import models as m
import roc_auc as ra
from prec_recall import pr, avg_pr
from prec_recall import create_pr, avg_pr
import sys
from sklearn.metrics import confusion_matrix
......@@ -187,7 +187,7 @@ def run_kfold(X_train, X_test, y_train, y_test, params):
cm+=confusion_matrix(y_test[i].argmax(axis=1), y_pred.argmax(axis=1))
#pr curve (contains 4 pr curves: one for each class)
recall, precision, average_prec = pr(N_CLASSES, y_test[i], y_pred)
recall, precision, average_prec = create_pr(N_CLASSES, y_test[i], y_pred)
recs_k.append(recall)
precs_k.append(precision)
avgs_k.append(average_prec)
......
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