[Plugin] Sklearn integration #2752
Labels
good first issue
Good for newcomers
hacktoberfest
plugins
Plugins related labels (backend or frontend)
Flyte is an ML- and data- aware orchestrator, it would make sense to support the most popular ML frameworks as first-class citizen types in the Flyte type system.
For
sklearn
, there's really only one base class, which is the BaseEstimator.We'll need to follow the conventions set in the
flytekit.extras.pytorch
module, by doing the following:__init__.py
file should handle the case where the user doesn't havesklearn
andjoblib
installed.TypeTransformer
for theBaseEstimator
class:BaseEstimators
Transformers
(e.g. https://scikit-learn.org/stable/modules/generated/sklearn.preprocessing.StandardScaler.html?highlight=standardscaler#sklearn.preprocessing.StandardScaler)sklearn.pipeline.Pipeline
Resources
Open Questions
x
and deserialized in another versiony
, but this is just a user warning... how do we handle this?The text was updated successfully, but these errors were encountered: