sklearn.utils.estimator_checks.parametrize_with_checks

sklearn.utils.estimator_checks.parametrize_with_checks()

Decorador específico de pytest para parametrizar las comprobaciones del estimador.

El id de cada comprobación se establece para que sea una versión de pprint del estimador y el nombre de la comprobación con sus argumentos de palabra clave. Esto permite usar pytest -k para especificar qué pruebas ejecutar:

pytest test_check_estimators.py -k check_estimators_fit_returns_self

Parámetros

estimatorslista de instancias de estimadores

Estimadores para generar comprobaciones.

Distinto en la versión 0.24: Pasar una clase quedó obsoleta en la versión 0.23, y el soporte para clases fue eliminado en 0.24.

Nuevo en la versión 0.24.

Devuelve

decorador : pytest.mark.parametrizepytest.mark.parametrize

Ejemplos

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

Ejemplos que usan sklearn.utils.estimator_checks.parametrize_with_checks

Aspectos Destacados de scikit-learn 0.22