Commit 0b91c7ca authored by lucas_miranda's avatar lucas_miranda
Browse files

Modified cluster purity computation. Instead of KNN, we now look at...

Modified cluster purity computation. Instead of KNN, we now look at neighborhoods of a predefined radius
parent 040daada
......@@ -95,7 +95,6 @@ class SEQ_2_SEQ_GMVAE(HyperModel):
self,
input_shape: tuple,
encoding: int,
entropy_reg_weight: float = 0.0,
kl_warmup_epochs: int = 0,
learn_rate: float = 1e-3,
loss: str = "ELBO+MMD",
......@@ -109,7 +108,6 @@ class SEQ_2_SEQ_GMVAE(HyperModel):
super().__init__()
self.input_shape = input_shape
self.encoding = encoding
self.entropy_reg_weight = entropy_reg_weight
self.kl_warmup_epochs = kl_warmup_epochs
self.learn_rate = learn_rate
self.loss = loss
......@@ -182,7 +180,6 @@ class SEQ_2_SEQ_GMVAE(HyperModel):
"units_lstm": lstm_units_1,
},
encoding=self.encoding,
entropy_reg_weight=self.entropy_reg_weight,
kl_warmup_epochs=self.kl_warmup_epochs,
loss=self.loss,
mmd_warmup_epochs=self.mmd_warmup_epochs,
......
......@@ -37,7 +37,6 @@ def test_SEQ_2_SEQ_AE_build(input_shape):
mmd_warmup_epochs=st.integers(min_value=0, max_value=5),
montecarlo_kl=st.integers(min_value=1, max_value=10),
number_of_components=st.integers(min_value=1, max_value=5),
entropy_reg_weight=st.one_of(st.just(0.0), st.just(1.0)),
)
def test_SEQ_2_SEQ_GMVAE_build(
loss,
......@@ -56,7 +55,6 @@ def test_SEQ_2_SEQ_GMVAE_build(
predictor=True,
phenotype_prediction=True,
overlap_loss=True,
entropy_reg_weight=entropy_reg_weight,
).build(
input_shape=(
100,
......
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