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)