Default split for mix at the end of function arguments

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