Commit 21861cd4 authored by Laurae's avatar Laurae Committed by Guolin Ke
Browse files

[Python-package]: Fix RandomState issue #376 (#377)

* Python: Fix RandomState issue #376

* Add test case for Python's Shuffle=True
parent 945bc150
...@@ -236,7 +236,7 @@ def _make_n_folds(full_data, data_splitter, nfold, params, seed, fpreproc=None, ...@@ -236,7 +236,7 @@ def _make_n_folds(full_data, data_splitter, nfold, params, seed, fpreproc=None,
folds = sfk.split(X=np.zeros(num_data), y=full_data.get_label()) folds = sfk.split(X=np.zeros(num_data), y=full_data.get_label())
else: else:
if shuffle: if shuffle:
randidx = np.RandomState(seed).random.permutation(num_data) randidx = np.random.RandomState(seed).permutation(num_data)
else: else:
randidx = np.arange(num_data) randidx = np.arange(num_data)
kstep = int(num_data / nfold) kstep = int(num_data / nfold)
......
...@@ -149,6 +149,9 @@ class TestEngine(unittest.TestCase): ...@@ -149,6 +149,9 @@ class TestEngine(unittest.TestCase):
lgb.cv({'verbose': -1}, lgb_train, num_boost_round=20, nfold=5, shuffle=False, lgb.cv({'verbose': -1}, lgb_train, num_boost_round=20, nfold=5, shuffle=False,
metrics='l1', verbose_eval=False, metrics='l1', verbose_eval=False,
callbacks=[lgb.reset_parameter(learning_rate=lambda i: 0.1 - 0.001 * i)]) callbacks=[lgb.reset_parameter(learning_rate=lambda i: 0.1 - 0.001 * i)])
lgb.cv({'verbose': -1}, lgb_train, num_boost_round=20, nfold=5, shuffle=True,
metrics='l1', verbose_eval=False,
callbacks=[lgb.reset_parameter(learning_rate=lambda i: 0.1 - 0.001 * i)])
tss = TimeSeriesSplit(3) tss = TimeSeriesSplit(3)
lgb.cv({'verbose': -1}, lgb_train, num_boost_round=20, data_splitter=tss, nfold=5, # test if wrong nfold is ignored lgb.cv({'verbose': -1}, lgb_train, num_boost_round=20, data_splitter=tss, nfold=5, # test if wrong nfold is ignored
metrics='l2', verbose_eval=False) metrics='l2', verbose_eval=False)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment