Commit ce65f935 authored by lucas_miranda's avatar lucas_miranda
Browse files

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

parent b6d2cd57
Pipeline #93322 failed with stage
in 26 minutes and 57 seconds
......@@ -278,7 +278,7 @@ project_coords = project(
animal_ids=tuple([animal_id]),
arena="circular",
arena_dims=tuple([arena_dims]),
enable_iterative_imputation=False,
enable_iterative_imputation=True,
exclude_bodyparts=exclude_bodyparts,
exp_conditions=treatment_dict,
path=train_path,
......@@ -350,7 +350,6 @@ if not tune:
trained_models = project_coords.deep_unsupervised_embedding(
(X_train, y_train, X_val, y_val),
epochs=1,
batch_size=batch_size,
encoding_size=encoding_size,
hparams=hparams,
......
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