Merge 97fc98a into adae7e0

.travis.yml

@@ -1,12 +1,14 @@
 language: python
+python:
+    - "3.5"
 virtualenv:
   system_site_packages: true
 env:
   matrix:
     # let's start simple:
-    - PYTHON_VERSION="2.7" LATEST="true"
-    - PYTHON_VERSION="3.6" COVERAGE="true" LATEST="true"
-    - PYTHON_VERSION="3.6" LATEST="true"
+    - PYTHON_VERSION="3.5" LATEST="true"
+    - PYTHON_VERSION="3.5" COVERAGE="true" LATEST="true"
+    - PYTHON_VERSION="3.5" LATEST="true"
 install: source ./ci/.travis_install.sh
 script: bash ./ci/.travis_test.sh
 after_success:

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=40, size_feature_subset=5, 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

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)------------------------------------------------------------