diff options
| -rw-r--r-- | lenet.py | 2 |
1 files changed, 1 insertions, 1 deletions
@@ -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 |