Skip to content

convert_sklearn_to_river

Wraps a scikit-learn estimator to make it compatible with river.

Parameters

  • estimator

    Typesklearn_base.BaseEstimator

  • classes

    Typelist[base.typing.ClfTarget] | None

    DefaultNone

    Class names. Required for classifiers, and not accepted for regressors.