Commit 8c2a744a authored by lucas_miranda's avatar lucas_miranda
Browse files

Fixed issue with gmvaep when returning both prior and posterior

parent 56bc7c7c
Pipeline #98190 canceled with stages
in 12 minutes and 14 seconds
......@@ -188,7 +188,7 @@ class SEQ_2_SEQ_GMVAE(HyperModel):
overlap_loss=self.overlap_loss,
phenotype_prediction=self.pheno_class,
predictor=self.predictor,
).build(self.input_shape)[-1]
).build(self.input_shape)[-3]
return gmvaep
......
......@@ -137,7 +137,7 @@ def test_autoencoder_fitting(
),
batch_size=st.integers(min_value=128, max_value=512),
encoding_size=st.integers(min_value=1, max_value=16),
hpt_type=st.one_of(st.just("bayopt"), st.just("hypermodel")),
hpt_type=st.one_of(st.just("bayopt"), st.just("hyperband")),
hypermodel=st.just("S2SGMVAE"),
k=st.integers(min_value=1, max_value=10),
loss=st.one_of(st.just("ELBO"), st.just("MMD")),
......
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