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

Parameters:

estimators估计器实例列表: 要生成检查的估计器。

Changed in version 0.24: 在版本0.23中传递类已被弃用，并且在0.24中删除了对类的支持。请传递一个实例。

Added in version 0.24.

Returns:

decorator :pytest.mark.parametrize

See also

check_estimator: 检查估计器是否符合scikit-learn约定。

Examples

>>> 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)

Gallery examples#

scikit-learn 0.22 版本发布亮点