pass integer random_state to estimator in imputer (scikit-learn#15611)

david-cortes · Nov 20, 2019 · 01b04c5 · 01b04c5
1 parent 3e4ad05
commit 01b04c5
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/sklearn/impute/_iterative.py b/sklearn/impute/_iterative.py
@@ -558,8 +558,11 @@ def fit_transform(self, X, y=None):
         else:
             self._estimator = clone(self.estimator)
 
-        if hasattr(self._estimator, 'random_state'):
-            self._estimator.random_state = self.random_state_
+        if hasattr(self._estimator, "random_state"):
+            if self._estimator.random_state is None:
+                random_state = self.random_state_.randint(
+                    np.iinfo(np.int32).max)
+                self._estimator.set_params(random_state = random_state)
 
         self.imputation_sequence_ = []