Commit 3cef046e authored by lucas_miranda's avatar lucas_miranda
Browse files

Added extra branch to main autoencoder for rule_based prediction

parent aafa39ee
......@@ -85,10 +85,10 @@ def test_get_callbacks(
@settings(max_examples=1, deadline=None, suppress_health_check=[HealthCheck.too_slow])
@given(
loss=st.one_of(st.just("ELBO"), st.just("MMD"), st.just("ELBO+MMD")),
next_sequence_prediction=st.one_of(st.just(1.0), st.just(1.0)),
phenotype_prediction=st.one_of(st.just(1.0), st.just(1.0)),
rule_based_prediction=st.one_of(st.just(1.0), st.just(1.0)),
variational=st.one_of(st.just(True), st.just(True)),
next_sequence_prediction=st.one_of(st.just(0.0), st.just(1.0)),
phenotype_prediction=st.one_of(st.just(0.0), st.just(1.0)),
rule_based_prediction=st.one_of(st.just(0.0), st.just(1.0)),
variational=st.one_of(st.just(True), st.just(False)),
)
def test_autoencoder_fitting(
loss,
......
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