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

fix bug

parent 0b3210d9
...@@ -16,7 +16,7 @@ def get_roc_curve(label,pred): ...@@ -16,7 +16,7 @@ def get_roc_curve(label,pred):
""" """
计算二分类问题的roc和auc 计算二分类问题的roc和auc
""" """
try:
test_label = pd.read_table(label) test_label = pd.read_table(label)
pred_label = pd.read_table(pred) pred_label = pd.read_table(pred)
y = test_label.values y = test_label.values
...@@ -34,10 +34,7 @@ def get_roc_curve(label,pred): ...@@ -34,10 +34,7 @@ def get_roc_curve(label,pred):
# plt.text(0.5,0.8,AUC,color='blue',ha='center') # plt.text(0.5,0.8,AUC,color='blue',ha='center')
# # plt.savefig(output) # # plt.savefig(output)
print(AUC) 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__": # 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