some corrections made

cnn.py

@@ -124,7 +124,7 @@ def run_nn(input_, output_, n_experiences, params):
 	print("Average accuracy: ", total_acc)
 	print("Average area under the curve: ", total_auc)
 
-	return total_acc, total_auc, X_train_kfold, X_test_kfold, validation_Y_one_hot
+	return total_acc, total_auc, X_train_kfold, X_test_kfold, train_Y_one_hot, validation_Y_one_hot
 
 
 def run_kfold(X_train, X_test, y_train, y_test, params):