diff --git a/train.py b/train.py index 59389ea..c8072dc 100644 --- a/train.py +++ b/train.py @@ -43,15 +43,15 @@ def harmonize(data): nodes_number = 17 while (nodes_number>=3): if nodes_number == 17: - layer = Dense(nodes_number, activation='tanh')(input_layer) + layer = Dense(nodes_number, activation='relu')(input_layer) else: - layer = Dense(nodes_number, activation='tanh')(layer) + layer = Dense(nodes_number, activation='relu')(layer) nodes_number=nodes_number-2 nodes_number=5 while(nodes_number<=17): - layer = Dense(nodes_number, activation='tanh')(layer) + layer = Dense(nodes_number, activation='relu')(layer) nodes_number=nodes_number+2 -output_layer = Dense(input_dim, activation='tanh')(layer) +output_layer = Dense(input_dim, activation='relu')(layer) autoencoder = Model(inputs=input_layer, outputs=output_layer) nb_epoch = 10