pr curve averaged for each class

main.py

@@ -43,7 +43,7 @@ for params in combinations:
 # once we have chosen the optimal parameters we can do the normal kfold
 
 #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)
+acc, auc, cm, pr = cnn.run_kfold(X_train_kfold_opt, X_test_kfold_opt, y_train_kfold_opt, y_test_kfold_opt, max_params)
 #to add: precision recall curve