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_s
参数 说明
estimators list of estimators objects or classes
用于生成检查的估计量
从0.23版开始不推荐使用:从0.23版开始不推荐使用传递类,因此0.24版不支持该类。 而是传递一个实例。
返回值 说明
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)

sklearn.utils.estimator_checks.parametrize_with_checks使用示例