26 lines
963 B
Python
26 lines
963 B
Python
"""The :mod:`sklearn.inspection` module includes tools for model inspection."""
|
|
|
|
# TODO: remove me in 0.24 (as well as the noqa markers) and
|
|
# import the partial_dependence func directly from the
|
|
# ._partial_dependence module instead.
|
|
# Pre-cache the import of the deprecated module so that import
|
|
# sklearn.inspection.partial_dependence returns the function as in
|
|
# 0.21, instead of the module
|
|
# https://github.com/scikit-learn/scikit-learn/issues/15842
|
|
import warnings
|
|
with warnings.catch_warnings():
|
|
warnings.simplefilter("ignore", category=FutureWarning)
|
|
from .partial_dependence import partial_dependence
|
|
|
|
from ._permutation_importance import permutation_importance # noqa
|
|
|
|
from ._plot.partial_dependence import plot_partial_dependence # noqa
|
|
from ._plot.partial_dependence import PartialDependenceDisplay # noqa
|
|
|
|
|
|
__all__ = [
|
|
'partial_dependence',
|
|
'plot_partial_dependence',
|
|
'permutation_importance',
|
|
'PartialDependenceDisplay'
|
|
]
|