From 0b8861d8f1ed02e45b3980758e304c873c28e3a3 Mon Sep 17 00:00:00 2001
From: name2965 <66663783+name2965@users.noreply.github.com>
Date: Thu, 8 Jun 2023 13:06:31 +0900
Subject: [PATCH] Update train.py

Change the activation function
---
 train.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

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