From 8f4d23e5adcedfd262123d3b84bc6c20cb7ab63f Mon Sep 17 00:00:00 2001
From: lucas_miranda <lucasmiranda42@gmail.com>
Date: Thu, 19 Nov 2020 19:14:52 +0100
Subject: [PATCH] Refactored a bit the old visualization pipeline - added umap
to requirements
---
examples/visualizations/train_viz_data_generator.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/examples/visualizations/train_viz_data_generator.py b/examples/visualizations/train_viz_data_generator.py
index 8841ffae..4878085d 100644
--- a/examples/visualizations/train_viz_data_generator.py
+++ b/examples/visualizations/train_viz_data_generator.py
@@ -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
--
GitLab