diff --git a/mlp.py b/mlp.py index 222225cd81538be3802564086da0e179efaafbbc..ec1a02eee8d866cbc6c8883b24592db2e478335b 100644 --- a/mlp.py +++ b/mlp.py @@ -152,7 +152,7 @@ def run_mlp_training(data_train, labels_train, data_test, labels_test, d_h, lear if __name__ == "__main__": # Parameters - split_factor = 0.1 + split_factor = 0.9 d_h = 64 learning_rate = 0.1 num_epoch = 100