Commit 1cd9d1b6 authored by lucas_miranda's avatar lucas_miranda
Browse files

Implemented KNN_purity callback

parent 552696d2
......@@ -253,7 +253,7 @@ class SEQ_2_SEQ_GMVAE:
montecarlo_kl: int = 1,
neuron_control: bool = False,
number_of_components: int = 1,
overlap_loss: float = -1,
overlap_loss: float = -1.,
phenotype_prediction: float = 0.0,
predictor: float = 0.0,
reg_cat_clusters: bool = False,
......@@ -606,7 +606,7 @@ class SEQ_2_SEQ_GMVAE:
z_gauss = deepof.model_utils.Cluster_overlap(
self.ENCODING,
self.number_of_components,
loss=tf.maximum(0, self.overlap_loss).numpy(),
loss=tf.maximum(0., self.overlap_loss).numpy(),
)(z_gauss)
z = tfpl.DistributionLambda(
......
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