sklearn.utils.estimator_checks.check_estimator

sklearn.utils.estimator_checks.check_estimator()

Comprueba si el estimador cumple las convenciones de scikit-learn.

Este estimador ejecutará un extenso conjunto de pruebas para la validación de la entrada, las formas, etc, asegurándose de que el estimador cumple con las convenciones de scikit-learn como se detalla en Rodando tu propio estimador. Se ejecutarán pruebas adicionales para clasificadores, regresores, conglomerados o transformadores si la clase Estimator hereda del mixin correspondiente de sklearn.base.

Al establecer generate_only=True se obtiene un generador que produce tuplas (estimador, comprobación) en las que la comprobación puede llamarse de forma independiente, es decir, check(estimador). Esto permite que todas las comprobaciones se ejecuten de forma independiente y que se informe de las comprobaciones que fallan.

scikit-learn proporciona un decorador específico para pytest, parametrize_with_checks, haciendo más fácil probar múltiples estimadores.

Parámetros

Estimadorobjeto estimador

Instancia del estimador a comprobar.

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

generate_onlybool, default=False

Cuando False, las comprobaciones son evaluadas cuando check_estimator es llamado. Cuando True, check_estimator devuelve un generador que yields (estimador, verificar) tuplas. El cheque se ejecuta llamando check(estimator).

Nuevo en la versión 0.22.

Devuelve

checks_generatorgenerador

Generador que produce tuplas (estimador, verificación). Se devuelve cuando generate_only=True.