ertk.sklearn.models.get_sk_model_fn

ertk.sklearn.models.get_sk_model_fn(name: str) → Callable[[...], BaseEstimator]

Get a scikit-learn model class.

Parameters:

name: str

The model name.

Returns:

model_fn: callable

A method that takes arguments and returns a Model instance.