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

Added latent regularization control to deepof.data.coordinates.deep_unsupervised_embedding()

parent 27daa823
Pipeline #93783 passed with stage
in 42 minutes and 11 seconds
......@@ -299,7 +299,7 @@ class project:
for k, value in tab_dict.items():
imputed = IterativeImputer(
max_iter=250, skip_complete=True
max_iter=1000, skip_complete=True
).fit_transform(value)
tab_dict[k] = pd.DataFrame(
imputed, index=value.index, columns=value.columns
......
......@@ -89,13 +89,13 @@ def load_treatments(train_path):
def get_callbacks(
X_train: np.array,
batch_size: int,
cp: bool,
variational: bool,
phenotype_class: float,
predictor: float,
loss: str,
reg_cat_clusters: bool,
reg_cluster_variance: bool,
cp: bool = False,
reg_cat_clusters: bool = False,
reg_cluster_variance: bool = False,
logparam: dict = None,
outpath: str = ".",
) -> List[Union[Any]]:
......
......@@ -77,13 +77,13 @@ def test_get_callbacks(
runID, tbc, cycle1c, cpc = deepof.train_utils.get_callbacks(
X_train,
batch_size,
True,
variational,
pheno_class,
predictor,
loss,
True,
True,
True,
None,
)
assert type(runID) == str
......@@ -177,11 +177,11 @@ def test_tune_search(
deepof.train_utils.get_callbacks(
X_train,
batch_size,
False,
hypermodel == "S2SGMVAE",
0,
predictor,
loss,
False,
True,
True,
None,
......
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