Uploaded image for project: 'Spark'
  1. Spark
  2. SPARK-4587

Model export/import

    XMLWordPrintableJSON

Details

    • Umbrella
    • Status: Resolved
    • Critical
    • Resolution: Fixed
    • None
    • 2.0.0
    • ML, MLlib
    • None

    Description

      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.

      UPDATE: Design doc for model import/export

      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).
      • 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.

      Attachments

        Issue Links

          Activity

            People

              josephkb Joseph K. Bradley
              mengxr Xiangrui Meng
              Votes:
              10 Vote for this issue
              Watchers:
              40 Start watching this issue

              Dates

                Created:
                Updated:
                Resolved: