Skip to content
Snippets Groups Projects
Commit 2ce1be9d authored by Lucas Miranda's avatar Lucas Miranda
Browse files

Added MinMaxScaler option to preprocessing

parent 17031fbc
No related branches found
No related tags found
No related merge requests found
...@@ -6,7 +6,7 @@ from tensorflow.keras.initializers import he_uniform, Orthogonal ...@@ -6,7 +6,7 @@ from tensorflow.keras.initializers import he_uniform, Orthogonal
from tensorflow.keras.layers import BatchNormalization, Bidirectional, Dense from tensorflow.keras.layers import BatchNormalization, Bidirectional, Dense
from tensorflow.keras.layers import Dropout, Lambda, LSTM from tensorflow.keras.layers import Dropout, Lambda, LSTM
from tensorflow.keras.layers import RepeatVector, TimeDistributed from tensorflow.keras.layers import RepeatVector, TimeDistributed
from tensorflow.keras.losses import BinaryCrossentropy, Huber from tensorflow.keras.losses import Huber
from tensorflow.keras.optimizers import Adam from tensorflow.keras.optimizers import Adam
from source.model_utils import * from source.model_utils import *
import tensorflow as tf import tensorflow as tf
...@@ -137,7 +137,7 @@ class SEQ_2_SEQ_AE: ...@@ -137,7 +137,7 @@ class SEQ_2_SEQ_AE:
model = Sequential([encoder, decoder], name="SEQ_2_SEQ_AE") model = Sequential([encoder, decoder], name="SEQ_2_SEQ_AE")
model.compile( model.compile(
loss="binary_crossentropy",#Huber(reduction="sum", delta=100.0), loss=Huber(reduction="sum", delta=100.0),
optimizer=Adam(lr=self.learn_rate, clipvalue=0.5,), optimizer=Adam(lr=self.learn_rate, clipvalue=0.5,),
metrics=["mae"], metrics=["mae"],
) )
...@@ -309,7 +309,7 @@ class SEQ_2_SEQ_VAE: ...@@ -309,7 +309,7 @@ class SEQ_2_SEQ_VAE:
return self.input_shape[1:] * huber(x_, x_decoded_mean_) return self.input_shape[1:] * huber(x_, x_decoded_mean_)
vae.compile( vae.compile(
loss="binary_crossentropy",#huber_loss, loss=huber_loss,
optimizer=Adam(lr=self.learn_rate,), optimizer=Adam(lr=self.learn_rate,),
metrics=["mae"], metrics=["mae"],
experimental_run_tf_function=False, experimental_run_tf_function=False,
...@@ -518,7 +518,7 @@ class SEQ_2_SEQ_VAEP: ...@@ -518,7 +518,7 @@ class SEQ_2_SEQ_VAEP:
return self.input_shape[1:] * huber(x_, x_decoded_mean_) return self.input_shape[1:] * huber(x_, x_decoded_mean_)
vaep.compile( vaep.compile(
loss="binary_crossentropy", loss=huber_loss,
optimizer=Adam(lr=self.learn_rate,), optimizer=Adam(lr=self.learn_rate,),
metrics=["mae"], metrics=["mae"],
experimental_run_tf_function=False, experimental_run_tf_function=False,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment