fix two bugs for classification

fix two bugs for classification

Author: taigw

pymic/net_run/agent_cls.py

@@ -308,7 +308,7 @@ def infer(self):
             csv_writer = csv.writer(csv_file, delimiter=',', 
                                 quotechar='"',quoting=csv.QUOTE_MINIMAL)
             head = ['image', 'label']
-            if(len(out_lab_list[0]) > 1):
+            if(len(out_lab_list[0]) > 2):
                 head = ['image'] + ['label{0:}'.format(i) for i in range(class_num)]
             csv_writer.writerow(head)
             for item in out_lab_list:

pymic/util/evaluation_cls.py

@@ -62,7 +62,7 @@ def binary_evaluation(config):
     for i in range(len(gt_items)):
         assert(gt_items.iloc[i, 0] == prob_items.iloc[i, 0])
 
-    gt_data  = np.asarray(gt_items.iloc[:, 1])
+    gt_data   = np.asarray(gt_items.iloc[:, -1])
     prob_data = np.asarray(prob_items.iloc[:, 1:])
     score_list = []
     for metric in metric_list: