Commit 61438b13 authored by 张彦钊's avatar 张彦钊

fix bug

parent 0b3210d9
......@@ -16,7 +16,7 @@ def get_roc_curve(label,pred):
"""
计算二分类问题的roc和auc
"""
try:
test_label = pd.read_table(label)
pred_label = pd.read_table(pred)
y = test_label.values
......@@ -34,10 +34,7 @@ def get_roc_curve(label,pred):
# plt.text(0.5,0.8,AUC,color='blue',ha='center')
# # plt.savefig(output)
print(AUC)
except:
print("the format of the file must be the n*1")
print("the test_label must be 0 or 1")
print("the test_pred must be at [0,1]")
#
# if __name__ == "__main__":
......
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