def test_sklearn_check_estimator(estimator): estimator_checks.check_estimator( compat.convert_creme_to_sklearn(estimator))
def test_creme_to_sklearn_check_estimator(estimator: base.Estimator): skl_estimator = compat.convert_creme_to_sklearn(estimator) estimator_checks.check_estimator(skl_estimator)
def test_sklearn_check_twoway(): estimator = sk_linear_model.SGDRegressor() creme_estimator = compat.convert_sklearn_to_creme(estimator) skl_estimator = compat.convert_creme_to_sklearn(creme_estimator) estimator_checks.check_estimator(skl_estimator)
def test_check_estimator(): model = linear_model.LinearRegression() skl_model = compat.convert_creme_to_sklearn(model) estimator_checks.check_estimator(skl_model)