Package org.apache.spark.ml.util

Interface MLWriterFormat

All Known Subinterfaces:
MLFormatRegister
All Known Implementing Classes:
InternalKMeansModelWriter, InternalLinearRegressionModelWriter, PMMLKMeansModelWriter, PMMLLinearRegressionModelWriter
public interface MLWriterFormat
Abstract class to be implemented by objects that provide ML exportability.

A new instance of this class will be instantiated each time a save call is made.

Must have a valid zero argument constructor which will be called to instantiate.

Since:
2.4.0

  • Method Summary

    Modifier and Type
    Method
    Description
    void
    write(String path, SparkSession session, scala.collection.mutable.Map<String,String> optionMap, PipelineStage stage)
    Function to write the provided pipeline stage out.

  • Method Details

    • write

      void write(String path, SparkSession session, scala.collection.mutable.Map<String,String> optionMap, PipelineStage stage)
      Function to write the provided pipeline stage out.

      Parameters:
      path - The path to write the result out to.
      session - SparkSession associated with the write request.
      optionMap - User provided options stored as strings.
      stage - The pipeline stage to be saved.