Commit 8f4d23e5 authored by lucas_miranda's avatar lucas_miranda
Browse files

Refactored a bit the old visualization pipeline - added umap to requirements

parent 443f54df
Pipeline #87203 passed with stage
in 20 minutes and 38 seconds
......@@ -216,7 +216,7 @@ parser.add_argument(
)
parser.add_argument(
"--samples",
"-sa",
"-sam",
help="Sets the number of samples (without replacement) to take from the validation set.",
default=5000,
type=int,
......@@ -237,7 +237,7 @@ animal_id = args.animal_id
arena_dims = args.arena_dims
batch_size = args.batch_size
bayopt_trials = args.bayopt
checkpoints = args.Zcheckpoint_path
checkpoints = args.checkpoint_path
exclude_bodyparts = tuple(args.exclude_bodyparts.split(","))
gaussian_filter = args.gaussian_filter
hparams = args.hyperparameters
......
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