Unknown type of parameter:random_state, got:RandomState (#78)

* Update boruta_py.py

remove check_random_state function for LightGBM and XGBoost to work

* Update boruta_py.py

additional control for random_state

* Update boruta_py.py

restoring original self.random_state = check_random_state(self.random_state)

* Update boruta_py.py

final fix for random_state problem with XGBoost and LightGBM

* Update boruta_py.py

fixed error message to original

* adding __version__ method

* updating estimator if a lightgbm, otherwise leaving it as it is

* removing _is_lightgbm function

Author: lmassaron

boruta/boruta_py.py

@@ -184,6 +184,8 @@ def __init__(self, estimator, n_estimators=1000, perc=100, alpha=0.05,
         self.max_iter = max_iter
         self.random_state = random_state
         self.verbose = verbose
+        self.__version__ = '0.3'
+        self._is_lightgbm = 'lightgbm' in str(type(self.estimator))
 
     def fit(self, X, y):
         """
@@ -301,7 +303,10 @@ def _fit(self, X, y):
                 self.estimator.set_params(n_estimators=n_tree)
 
             # make sure we start with a new tree in each iteration
-            self.estimator.set_params(random_state=self.random_state)
+            if self.is_lgb:
+                self.estimator.set_params(random_state=self.random_state.randint(0, 10000))
+            else:
+                self.estimator.set_params(random_state=self.random_state)
 
             # add shadow attributes, shuffle them and train estimator, get imps
             cur_imp = self._add_shadows_get_imps(X, y, dec_reg)