Commit eee28cf0 authored by lucas_miranda's avatar lucas_miranda
Browse files

Updated default settings in test_data.py

parent 7ebbdb96
Pipeline #95866 passed with stages
in 19 minutes and 17 seconds
......@@ -870,7 +870,6 @@ class coordinates:
variational: bool = True,
reg_cat_clusters: bool = False,
reg_cluster_variance: bool = False,
entropy_radius: float = 0.75,
entropy_samples: int = 10000,
) -> Tuple:
"""
......@@ -932,7 +931,6 @@ class coordinates:
variational=variational,
reg_cat_clusters=reg_cat_clusters,
reg_cluster_variance=reg_cluster_variance,
entropy_radius=entropy_radius,
entropy_samples=entropy_samples,
)
......
......@@ -284,7 +284,7 @@ class neighbor_cluster_purity(tf.keras.callbacks.Callback):
neighbor_number[i] = np.sum(neighborhood)
# Compute weights multiplying neighbor number and target confidence
purity_weights = neighbor_number * max_groups
purity_weights = neighbor_number * max_groups[random_idxs]
writer = tf.summary.create_file_writer(self.log_dir)
with writer.as_default():
......
......@@ -57,16 +57,17 @@ def test_get_callbacks(
loss,
):
runID, tbc, entropy, cycle1c, cpc = deepof.train_utils.get_callbacks(
X_train,
batch_size,
variational,
pheno_class,
predictor,
loss,
True,
True,
True,
None,
X_train=X_train,
batch_size=batch_size,
variational=variational,
phenotype_class=pheno_class,
predictor=predictor,
loss=loss,
X_val=X_train,
cp=True,
reg_cat_clusters=False,
reg_cluster_variance=False,
logparam={"encoding": 2, "k": 15},
)
assert type(runID) == str
assert type(tbc) == tf.keras.callbacks.TensorBoard
......@@ -117,7 +118,6 @@ def test_autoencoder_fitting(
phenotype_class=pheno_class,
predictor=predictor,
variational=variational,
entropy_radius=0.75,
entropy_samples=10,
)
......@@ -171,7 +171,7 @@ def test_tune_search(
reg_cat_clusters=True,
reg_cluster_variance=True,
entropy_samples=10,
logparam=None,
logparam={"encoding": 2, "k": 15},
)
)[1:]
......
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