Skip to main content

5 docs tagged with "sklearn"

View All Tags

Grid Search with Metaflow

I want to do a grid search with Metaflow. How can I use ParameterGrid and GridSearchCV with Metaflow?

Use Scikit-learn Estimators with Metaflow

I have a Scikit-learn workflow that I want to incorporate into a Metaflow flow. How can I include model fitting, prediction, feature transformations, and other capabilities enabled by Scikit-learn in flow steps?