1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
|
"""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'
]
|
