Commit c22cd53d by Pamela Osuna

just deleted something associated to the roc avg curve

parent fed1cb40
...@@ -393,7 +393,7 @@ class CNN_Antifrag: ...@@ -393,7 +393,7 @@ class CNN_Antifrag:
precs_k.append(precision) precs_k.append(precision)
avgs_k.append(average_prec) avgs_k.append(average_prec)
#auc #auc
fpr, tpr, auc = ra.roc_auc(N, y_test, y_pred) fpr, tpr, auc = ra.roc_auc(N, y_test, y_pred)
# calculate area under the curve # calculate area under the curve
total_auc += auc total_auc += auc
...@@ -406,4 +406,4 @@ class CNN_Antifrag: ...@@ -406,4 +406,4 @@ class CNN_Antifrag:
print("Average accuracy: ", total_acc) print("Average accuracy: ", total_acc)
print("Average area under the curve: ", total_auc) print("Average area under the curve: ", total_auc)
return total_acc, total_auc, cm, pr, roc return total_acc, total_auc, cm, pr
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