pr and cm

main.py

@@ -44,9 +44,11 @@ for params in combinations:
 
 #note: the test data remains unbalanced
 acc, auc, cm = cnn.run_kfold(X_train_opt, X_test_kfold_opt, y_train_kfold_opt, y_test_kfold_opt, max_params)
+#to add: precision recall curve
+
 
 labels = ['~robust&~evolvable', 'evolvable&~robust', 'robust&~evolvable', 'robust&evolvable']
-plot_confusion_matrix(cm, labels)
+plot_confusion_matrix(cm, labels) #this function saves the matrix image automatically
 
 f = open("acc_auc.txt", 'w+')
 f.write("Average accuracy: " + str(acc)+"\n")