corrections

cnn.py

@@ -113,8 +113,8 @@ def run_nn(input_, output_, n_experiences, params):
 
 
 
 
 	        # calculate area under the curve and confu
 	        # calculate area under the curve and confu
-	        y_pred = model.predict(X_validation, batch_size = bs)
-	        fpr, tpr, auc = ra.roc_auc(N_CLASSES, validation_Y_one_hot, y_pred)
+	        y_pred = model.predict(X_validation[i][j], batch_size = bs)
+	        fpr, tpr, auc = ra.roc_auc(N_CLASSES, validation_Y_one_hot[i][j], y_pred)
 	        total_auc += auc
 	        total_auc += auc
 	        print("Area under the curve:", auc)
 	        print("Area under the curve:", auc)
 
 
@@ -171,7 +171,7 @@ def run_kfold(X_train, X_test, y_train, y_test, params):
 
 
 
 
 		# calculate area under the curve
 		# calculate area under the curve
-		y_pred = model.predict(X_validation, batch_size = bs)
+		y_pred = model.predict(X_test, batch_size = bs)
 		fpr, tpr, auc = ra.roc_auc(N_CLASSES, y_test, y_pred)
 		fpr, tpr, auc = ra.roc_auc(N_CLASSES, y_test, y_pred)
 		total_auc += auc
 		total_auc += auc
 		print("Area under the curve:", auc)
 		print("Area under the curve:", auc)