Skip to content
Snippets Groups Projects
Commit 16a04429 authored by Audard Lucile's avatar Audard Lucile
Browse files

Update mlp.py

parent 65be793b
No related merge requests found
...@@ -152,7 +152,7 @@ def run_mlp_training(data_train, labels_train, data_test, labels_test, d_h, lear ...@@ -152,7 +152,7 @@ def run_mlp_training(data_train, labels_train, data_test, labels_test, d_h, lear
if __name__ == "__main__": if __name__ == "__main__":
# Parameters # Parameters
split_factor = 0.1 split_factor = 0.9
d_h = 64 d_h = 64
learning_rate = 0.1 learning_rate = 0.1
num_epoch = 100 num_epoch = 100
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment