fix bug

eda/ml_tools/rocCurve.py

@@ -2,14 +2,14 @@ import pandas as pd
 from sklearn import metrics
 # import matplotlib.pyplot as plt
 from sklearn.metrics import auc
-import argparse
+# import argparse
 
 
-parser = argparse.ArgumentParser()
-parser.add_argument('test_label',help='The filename of the test_label')
-parser.add_argument('test_pred',help='The filename of the test_pred')
-# parser.add_argument('output_photo',help='The filename of the output_photo')
-args = parser.parse_args()
+# parser = argparse.ArgumentParser()
+# parser.add_argument('test_label',help='The filename of the test_label')
+# parser.add_argument('test_pred',help='The filename of the test_pred')
+# # parser.add_argument('output_photo',help='The filename of the output_photo')
+# args = parser.parse_args()
 
 
 def get_roc_curve(label,pred):
@@ -39,6 +39,6 @@ def get_roc_curve(label,pred):
 		print("the test_label must be 0 or 1")
 		print("the test_pred must be at [0,1]")
 
-
-if __name__ == "__main__":
-	get_roc_curve(args.test_label,args.test_pred)
+#
+# if __name__ == "__main__":
+# 	get_roc_curve(args.test_label,args.test_pred)