This is an umbrella JIRA for one of the most requested features on the user mailing list. Model export/import can be done via Java serialization. But it doesn't work for models stored distributively, e.g., ALS and LDA. Ideally, we should provide save/load methods to every model. PMML is an option but it has its limitations. There are couple things we need to discuss: 1) data format, 2) how to preserve partitioning, 3) data compatibility between versions and language APIs, etc.
This document sketches machine learning model import/export plans, including goals, an API, and development plans.
UPDATE: As in the design doc, we plan to support:
- Our own Spark-specific format.
- This is needed to (a) support distributed models and (b) get model import/export support into Spark quickly (while avoiding the complexity of PMML).
- This is needed since it is the most commonly used format in industry.
This JIRA will be for the internal Spark-specific format described in the design doc. Parallel JIRAs will cover PMML.