Python sklearn parametrize_with_checks用法及代码示例

本文简要介绍python语言中 sklearn.utils.estimator_checks.parametrize_with_checks 的用法。

用法:

sklearn.utils.estimator_checks.parametrize_with_checks(estimators)

用于参数化估计器检查的 Pytest 特定装饰器。

每个检查的id 设置为估计器的 pprint 版本和检查的名称及其关键字参数。这允许使用 pytest -k 来指定要运行的测试：

pytest test_check_estimators.py -k check_estimators_fit_returns_self

参数：

estimators：估算器实例列表

生成检查的估计器。

返回：

decorator：pytest.mark.parametrize

例子：

>>> from sklearn.utils.estimator_checks import parametrize_with_checks
>>> from sklearn.linear_model import LogisticRegression
>>> from sklearn.tree import DecisionTreeRegressor

>>> @parametrize_with_checks([LogisticRegression(),
...                           DecisionTreeRegressor()])
... def test_sklearn_compatible_estimator(estimator, check):
...     check(estimator)