public class PMMLLinearRegressionModelWriter extends Object implements MLWriterFormat, MLFormatRegister
Constructor and Description |
---|
PMMLLinearRegressionModelWriter() |
Modifier and Type | Method and Description |
---|---|
String |
format()
The string that represents the format that this format provider uses.
|
String |
stageName()
The string that represents the stage type that this writer supports.
|
void |
write(String path,
SparkSession sparkSession,
scala.collection.mutable.Map<String,String> optionMap,
PipelineStage stage)
Function to write the provided pipeline stage out.
|
equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
shortName
public String format()
MLFormatRegister
override def format(): String =
"pmml"
Indicates that this format is capable of saving a pmml model.
Must have a valid zero argument constructor which will be called to instantiate.
Format discovery is done using a ServiceLoader so make sure to list your format in META-INF/services.
format
in interface MLFormatRegister
public String stageName()
MLFormatRegister
override def stageName(): String =
"org.apache.spark.ml.regression.LinearRegressionModel"
Indicates that this format is capable of saving Spark's own PMML model.
Format discovery is done using a ServiceLoader so make sure to list your format in META-INF/services.
stageName
in interface MLFormatRegister
public void write(String path, SparkSession sparkSession, scala.collection.mutable.Map<String,String> optionMap, PipelineStage stage)
MLWriterFormat
write
in interface MLWriterFormat
path
- The path to write the result out to.sparkSession
- SparkSession associated with the write request.optionMap
- User provided options stored as strings.stage
- The pipeline stage to be saved.