Commit aa625a5b authored by 张彦钊's avatar 张彦钊

fix a bug

parent 76fdbb84
......@@ -6,5 +6,5 @@ from config import *
if __name__ == "__main__":
test = pd.read_csv(DIRECTORY_PATH + "test.csv",header = None)
test_label = test[0].apply(lambda x: x[0]).values
predict = pd.read_csv(DIRECTORY_PATH + "model.out",header = None)[0].values
predict = pd.read_csv(DIRECTORY_PATH + "output.txt",header = None)[0].values
get_roc_curve(test_label,predict)
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