diff --git a/.travis.yml b/.travis.yml index 1123185..20d2413 100644 --- a/.travis.yml +++ b/.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: diff --git a/skrebate/vlsrelief.py b/skrebate/vlsrelief.py index 270fa66..7ade9ec 100644 --- 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=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 diff --git a/tests.py b/tests.py index 87561a4..4ec8519 100644 --- 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)------------------------------------------------------------