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 and 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 Detail

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.