Commit 51ca0a35 authored by lucas_miranda's avatar lucas_miranda
Browse files

Fixed issue with gmvaep when returning both prior and posterior

parent 3659d965
Pipeline #98191 canceled with stages
in 12 minutes and 13 seconds
......@@ -328,7 +328,14 @@ def autoencoder_fitting(
return_list = (encoder, decoder, ae)
else:
(encoder, generator, grouper, ae, prior, posterior) = deepof.models.SEQ_2_SEQ_GMVAE(
(
encoder,
generator,
grouper,
ae,
prior,
posterior,
) = deepof.models.SEQ_2_SEQ_GMVAE(
architecture_hparams=({} if hparams is None else hparams),
batch_size=batch_size,
compile_model=True,
......@@ -344,7 +351,9 @@ def autoencoder_fitting(
predictor=predictor,
reg_cat_clusters=reg_cat_clusters,
reg_cluster_variance=reg_cluster_variance,
).build(X_train.shape)
).build(
X_train.shape
)
return_list = (encoder, generator, grouper, ae)
if pretrained:
......
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