aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--lenet.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/lenet.py b/lenet.py
index ff042f1..fa27422 100644
--- a/lenet.py
+++ b/lenet.py
@@ -127,7 +127,7 @@ def test_classifier(model, x_test, y_true):
print("Test acc:", accuracy_score(y_true, y_pred))
plot_example_errors(y_pred, y_true, x_test)
-def mix_data(split=0, X_train, y_train, X_validation, y_validation, train_gen, tr_labels_gen, val_gen, val_labels_gen):
+def mix_data(X_train, y_train, X_validation, y_validation, train_gen, tr_labels_gen, val_gen, val_labels_gen, split=0):
if split == 0:
train_data = X_train