Save new sklearn wrapped models

Description

I am trying to save a model from one of the new sklearn compatible automl classes and am having difficulty doing so. I'd like to save the resulting trained model and then load it to make predictions in a production environment.

For example, here is my mlt object, an sklearn Pipeline:

Try pickle dump of everything:

Results in: TypeError: can't pickle dict_keys objects

Try h2o save of just the estimator:

Results in:

Assignee

Sebastien Poirier

Fix versions

None

Reporter

Stan Biryukov

Support ticket URL

None

Labels

Affected Spark version

None

Customer Request Type

None

Task progress

None

CustomerVisible

Yes

Components

Affects versions

Priority

Major
Configure