Handle 0 examples error only when neccessary

wenhui-prudencemed · Jan 17, 2017 · 0a90aa8 · 0a90aa8
1 parent cb61d85
commit 0a90aa8
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/ecg/eval-keras.py b/ecg/eval-keras.py
@@ -25,14 +25,15 @@
     predictions = np.load(open(args.prediction_path, 'rb'))
 
     y_val_flat = np.argmax(y_val, axis=-1).flatten().tolist()
-    y_val_flat.extend(range(len(dl.classes)))
     predictions_flat = np.argmax(predictions, axis=-1).flatten().tolist()
-    predictions_flat.extend(range(len(dl.classes)))
 
     print(classification_report(
         y_val_flat, predictions_flat,
         target_names=dl.classes))
 
+    y_val_flat.extend(range(len(dl.classes)))
+    predictions_flat.extend(range(len(dl.classes)))
+
     cnf_matrix = confusion_matrix(y_val_flat, predictions_flat).tolist()
     for i, row in enumerate(cnf_matrix):
         row.insert(0, dl.classes[i])