Disable shuffle for custom CV (#659)

* Disable shuffle for custom CV

* Add custom fold shuffle test

* Update test_split.py

* Update test_split.py
This commit is contained in:
jmrichardson
2022-08-12 20:05:32 -04:00
committed by GitHub
parent ca9f9054e7
commit e43485607a
3 changed files with 7 additions and 2 deletions

View File

@@ -174,6 +174,11 @@ def test_object():
automl._state.eval_method == "cv"
), "eval_method must be 'cv' for custom data splitter"
kf = TestKFold(5)
kf.shuffle = True
automl_settings["split_type"] = kf
automl.fit(X, y, **automl_settings)
if __name__ == "__main__":
test_groups()