fix bug joblib

AxeldeRomblay · Apr 26, 2019 · e360544 · e360544
1 parent 96c50e9
commit e360544
Show file tree

Hide file tree

Showing 6 changed files with 287 additions and 92 deletions.
diff --git a/.idea/workspace.xml b/.idea/workspace.xml
diff --git a/README.rst b/README.rst
@@ -81,7 +81,7 @@ MLBox main package contains 3 sub-packages : **preprocessing**, **optimisation**
            'fs__strategy' : {"space" : ["variance", "rf_feature_importance"]},
            'fs__threshold': {"search" : "choice", "space" : [0.1, 0.2, 0.3]},             
 
-           'est__strategy' : {"space" : ["XGBoost"]},
+           'est__strategy' : {"space" : ["LightGBM"]},
            'est__max_depth' : {"search" : "choice", "space" : [5,6]},
            'est__subsample' : {"search" : "uniform", "space" : [0.6,0.9]}
            

diff --git a/docs/conf.py b/docs/conf.py
@@ -42,8 +42,7 @@ def __getattr__(cls, name):
                 'sklearn.pipeline',
                 'sklearn.preprocessing',
                 'theano',
-                'lightgbm',
-                'xgboost'
+                'lightgbm'
                 ]
 
 sys.modules.update((mod_name, Mock()) for mod_name in MOCK_MODULES)

diff --git a/docs/introduction.rst b/docs/introduction.rst
@@ -60,7 +60,7 @@ MLBox main package contains 3 sub-packages : **preprocessing**, **optimisation**
            'fs__strategy' : {"space" : ["variance", "rf_feature_importance"]},
            'fs__threshold': {"search" : "choice", "space" : [0.1, 0.2, 0.3]},             
 
-           'est__strategy' : {"space" : ["XGBoost"]},
+           'est__strategy' : {"space" : ["LightGBM"]},
            'est__max_depth' : {"search" : "choice", "space" : [5,6]},
            'est__subsample' : {"search" : "uniform", "space" : [0.6,0.9]}
            

diff --git a/python-package/mlbox/__init__.py b/python-package/mlbox/__init__.py
@@ -2,7 +2,7 @@
 
 __author__ = """Axel ARONIO DE ROMBLAY"""
 __email__ = 'axelderomblay@gmail.com'
-__version__ = '0.6.0'
+__version__ = '0.6.1'
 
 from .preprocessing import *
 from .encoding import *

diff --git a/python-package/requirements.txt b/python-package/requirements.txt
@@ -4,6 +4,7 @@ hyperopt==0.1
 Keras==2.1.2
 pandas==0.21.0
 scikit-learn==0.19.0
+joblib==0.13.2
 Theano==1.0.1
 lightgbm==2.0.11
 networkx==1.11