fixed vls tests

EpistasisLab · Feb 12, 2020 · 97fc98a · 97fc98a
1 parent 1fa145c
commit 97fc98a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/skrebate/vlsrelief.py b/skrebate/vlsrelief.py
@@ -26,7 +26,7 @@ class VLSRelief(BaseEstimator, TransformerMixin):
 
     """
 
-    def __init__(self, core_algorithm, n_features_to_select=2, n_neighbors=100, num_feature_subset=30, size_feature_subset=10, discrete_threshold=10, verbose=False, n_jobs=1):
+    def __init__(self, core_algorithm, n_features_to_select=2, n_neighbors=100, num_feature_subset=20, size_feature_subset=10, discrete_threshold=10, verbose=False, n_jobs=1):
         """Sets up VLSRelief to perform feature selection.
 
         Parameters

diff --git a/tests.py b/tests.py
@@ -142,8 +142,8 @@ def test_vlsrelief_init():
     assert clf.discrete_threshold == 20
     assert clf.verbose == True
     assert clf.n_jobs == 3
-    assert clf.num_feature_subset == 40
-    assert clf.size_feature_subset == 5
+    assert clf.num_feature_subset == 20
+    assert clf.size_feature_subset == 10
 
 
 # Basic Parallelization Tests and Core binary data and discrete feature data testing (Focus on ReliefF only for efficiency)------------------------------------------------------------