Commit 32a2aadb authored by lucas_miranda's avatar lucas_miranda
Browse files

Drafted skeleton of SEQ2SEQ_VAEP (Variational AutoEncoding Predictor) in models.py

parent f61ed1fa
This diff is collapsed.
......@@ -217,7 +217,11 @@ class SEQ_2_SEQ_VAE:
)
# Decoder layers
Model_B1 = BatchNormalization()
Model_B2 = BatchNormalization()
Model_B3 = BatchNormalization()
Model_B4 = BatchNormalization()
Model_B5 = BatchNormalization()
Model_D0 = DenseTranspose(
Model_E5, activation="relu", output_dim=self.ENCODING,
)
......@@ -269,15 +273,15 @@ class SEQ_2_SEQ_VAE:
# Define and instanciate generator
generator = Model_D0(z)
generator = BatchNormalization()(generator)
generator = Model_B1(generator)
generator = Model_D1(generator)
generator = BatchNormalization()(generator)
generator = Model_B2(generator)
generator = Model_D2(generator)
generator = BatchNormalization()(generator)
generator = Model_B3(generator)
generator = Model_D3(generator)
generator = BatchNormalization()(generator)
generator = Model_B4(generator)
generator = Model_D4(generator)
generator = BatchNormalization()(generator)
generator = Model_B5(generator)
generator = Model_D5(generator)
x_decoded_mean = TimeDistributed(Dense(self.input_shape[2]))(generator)
......@@ -288,15 +292,15 @@ class SEQ_2_SEQ_VAE:
# Build generator as a separate entity
g = Input(shape=self.ENCODING)
_generator = Model_D0(g)
_generator = BatchNormalization()(_generator)
_generator = Model_B1(_generator)
_generator = Model_D1(_generator)
_generator = BatchNormalization()(_generator)
_generator = Model_B2(_generator)
_generator = Model_D2(_generator)
_generator = BatchNormalization()(_generator)
_generator = Model_B3(_generator)
_generator = Model_D3(_generator)
_generator = BatchNormalization()(_generator)
_generator = Model_B4(_generator)
_generator = Model_D4(_generator)
_generator = BatchNormalization()(_generator)
_generator = Model_B5(_generator)
_generator = Model_D5(_generator)
_x_decoded_mean = TimeDistributed(Dense(self.input_shape[2]))(_generator)
generator = Model(g, _x_decoded_mean, name="SEQ_2_SEQ_VGenerator")
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment