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)