Allow using RandomState object from Numpy in sklearn interface. (#5049)

2019-11-19 10:56:39 +08:00
parent 4d2779663e
commit a4f5c86276
2 changed files with 7 additions and 0 deletions
--- a/tests/python/test_with_sklearn.py
+++ b/tests/python/test_with_sklearn.py
@@ -450,6 +450,10 @@ def test_sklearn_random_state():
    clf = xgb.XGBClassifier(random_state=401)
    assert clf.get_xgb_params()['random_state'] == 401

+    random_state = np.random.RandomState(seed=403)
+    clf = xgb.XGBClassifier(random_state=random_state)
+    assert isinstance(clf.get_xgb_params()['random_state'], int)
+

 def test_sklearn_n_jobs():
    clf = xgb.XGBClassifier(n_jobs=1)