Commit 7b939a3b authored by lucas_miranda's avatar lucas_miranda
Browse files

Implemented KNN_purity callback

parent a07edada
Pipeline #95546 canceled with stages
in 11 minutes and 23 seconds
......@@ -160,18 +160,19 @@ def test_tune_search(
):
callbacks = list(
deepof.train_utils.get_callbacks(
X_train,
batch_size,
hypermodel == "S2SGMVAE",
0,
predictor,
loss,
False,
True,
True,
None,
X_train=X_train,
batch_size=batch_size,
variational=(hypermodel == "S2SGMVAE"),
phenotype_class=0,
predictor=predictor,
loss=loss,
X_val=X_train,
cp=False,
reg_cat_clusters=True,
reg_cluster_variance=True,
knn_neighbors=10,
knn_samples=10,
logparam=None,
)
)[1:]
......
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