Commit f1f49bc1 authored by lucas_miranda's avatar lucas_miranda
Browse files

Added extra branch to main autoencoder for rule_based prediction

parent 1bb7d1b8
Pipeline #99736 passed with stages
in 18 minutes and 59 seconds
......@@ -341,11 +341,11 @@ class SEQ_2_SEQ_GMVAE:
"bidirectional_merge": "concat",
"clipvalue": 1.0,
"dense_activation": "relu",
"dense_layers_per_branch": 3,
"dense_layers_per_branch": 1,
"dropout_rate": 0.05,
"learning_rate": 1e-3,
"units_conv": 64,
"units_dense2": 32,
"units_dense2": 64,
"units_lstm": 128,
}
......@@ -399,15 +399,23 @@ class SEQ_2_SEQ_GMVAE:
use_bias=True,
)
# Nested list comprehension adding a series of dense and batch norm layers
Model_E4 = [
Dense(
self.DENSE_2,
activation=self.dense_activation,
# kernel_constraint=UnitNorm(axis=0),
kernel_initializer=he_uniform(),
use_bias=True,
)
for _ in range(self.dense_layers_per_branch)
i
for s in [
[
Dense(
self.DENSE_2,
activation=self.dense_activation,
# kernel_constraint=UnitNorm(axis=0),
kernel_initializer=he_uniform(),
use_bias=True,
),
BatchNormalization(),
]
for _ in range(self.dense_layers_per_branch)
]
for i in s
]
# Decoder layers
......
......@@ -160,7 +160,7 @@
{
"data": {
"application/vnd.jupyter.widget-view+json": {
"model_id": "85438cadcf6e418b86e7174146f4dd1a",
"model_id": "ac7f3fee0d24497cbb92e55523fc9a22",
"version_major": 2,
"version_minor": 0
},
......@@ -257,13 +257,13 @@
},
{
"cell_type": "code",
"execution_count": 12,
"execution_count": 13,
"metadata": {},
"outputs": [
{
"data": {
"application/vnd.jupyter.widget-view+json": {
"model_id": "57f74bde88e049a98480291d8d83294f",
"model_id": "c9c967e149fc46eb91f7e90fd141a20f",
"version_major": 2,
"version_minor": 0
},
......
......@@ -283,6 +283,9 @@ def test_neighbor_latent_entropy():
X,
X,
callbacks=deepof.model_utils.neighbor_latent_entropy(
k=10, encoding_dim=6, validation_data=X, variational=True,
k=10,
encoding_dim=6,
validation_data=X,
variational=True,
),
)
Markdown is supported
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