Fix test_1d_feat_tsne

LukasHedegaard · Apr 17, 2023 · adcc7ac · adcc7ac
1 parent 6594eb8
commit adcc7ac
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/tests/dummy_dataset.py b/tests/dummy_dataset.py
@@ -137,21 +137,21 @@ def __init__(self, hparams):
         self.classes = ["fee", "fei"]
 
         self._train_dataloader = DataLoader(
-            OverOneDataset(self.input_shape, num_samples=65),
+            OverOneDataset(self.input_shape, num_samples=60),
             batch_size=hparams.batch_size,
             num_workers=num_workers,
             shuffle=True,
             pin_memory=num_workers > 1,
         )
         self._val_dataloader = DataLoader(
-            OverOneDataset(self.input_shape, num_samples=4),
+            OverOneDataset(self.input_shape, num_samples=30),
             batch_size=hparams.batch_size,
             num_workers=num_workers,
             shuffle=False,
             pin_memory=num_workers > 1,
         )
         self._test_dataloader = DataLoader(
-            OverOneDataset(self.input_shape, num_samples=10),
+            OverOneDataset(self.input_shape, num_samples=30),
             batch_size=hparams.batch_size,
             num_workers=num_workers,
             shuffle=False,

diff --git a/tests/test_feature_visualisation.py b/tests/test_feature_visualisation.py
@@ -131,7 +131,7 @@ def test_2d_feat_pca(self, caplog, main_and_args: Tuple[Main, AttributeDict]):
         assert is_nonempty_file(d / "features" / "test" / "conv_pca.npy")
         assert is_nonempty_file(d / "figures" / "test" / "conv_pca.png")
 
-    def test_1d_feat_tse(self, caplog, main_and_args: Tuple[Main, AttributeDict]):
+    def test_1d_feat_tsne(self, caplog, main_and_args: Tuple[Main, AttributeDict]):
         m, args = main_and_args
         args.test = True
         args.extract_features_after_layer = "lin"