You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
36 lines
1.3 KiB
36 lines
1.3 KiB
"""Enables Successive Halving search-estimators
|
|
|
|
The API and results of these estimators might change without any deprecation
|
|
cycle.
|
|
|
|
Importing this file dynamically sets the
|
|
:class:`~sklearn.model_selection.HalvingRandomSearchCV` and
|
|
:class:`~sklearn.model_selection.HalvingGridSearchCV` as attributes of the
|
|
`model_selection` module::
|
|
|
|
>>> # explicitly require this experimental feature
|
|
>>> from sklearn.experimental import enable_halving_search_cv # noqa
|
|
>>> # now you can import normally from model_selection
|
|
>>> from sklearn.model_selection import HalvingRandomSearchCV
|
|
>>> from sklearn.model_selection import HalvingGridSearchCV
|
|
|
|
|
|
The ``# noqa`` comment comment can be removed: it just tells linters like
|
|
flake8 to ignore the import, which appears as unused.
|
|
"""
|
|
|
|
# Authors: The scikit-learn developers
|
|
# SPDX-License-Identifier: BSD-3-Clause
|
|
|
|
from sklearn import model_selection
|
|
from sklearn.model_selection._search_successive_halving import (
|
|
HalvingGridSearchCV,
|
|
HalvingRandomSearchCV,
|
|
)
|
|
|
|
# use settattr to avoid mypy errors when monkeypatching
|
|
setattr(model_selection, "HalvingRandomSearchCV", HalvingRandomSearchCV)
|
|
setattr(model_selection, "HalvingGridSearchCV", HalvingGridSearchCV)
|
|
|
|
model_selection.__all__ += ["HalvingRandomSearchCV", "HalvingGridSearchCV"]
|