mirror of
https://github.com/microsoft/autogen.git
synced 2026-02-17 01:31:26 -05:00
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:
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user