Update pytests to specify number of trees in iForest. (#59,#14)

nasaharvest · Oct 4, 2021 · a07d8ee · a07d8ee
1 parent 362ccee
commit a07d8ee
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/test/planetary.config b/test/planetary.config
@@ -26,7 +26,9 @@ outlier_detection: {
     demud: {
         k: 5
     },
-    iforest: { },
+    iforest: {
+        n_trees: 100
+    },
     negative_sampling: {
         percent_increase: 20
     },

diff --git a/...functional/iforest/selections-iforest.csv → ...forest-n_trees=100/selections-iforest.csv b/...functional/iforest/selections-iforest.csv → ...forest-n_trees=100/selections-iforest.csv
diff --git a/test/test_planetary.py b/test/test_planetary.py
@@ -31,7 +31,7 @@ def test_run(self):
 
         # Check results for each algorithm
         for file_base in ['demud-k=5/selections-demud.csv',
-                          'iforest/selections-iforest.csv',
+                          'iforest-n_trees=100/selections-iforest.csv',
                           'lrx-inner_window=3-outer_window=9-bands=1/selections-lrx.csv',
                           'negative_sampling-percent_increase=20/selections-negative_sampling.csv',
                           'pca-k=5/selections-pca.csv'