MLWriter#

classpyspark.ml.util.MLWriter[source]#

Utility class that can save ML instances.

New in version 2.0.0.

Methods

option(key, value)	Adds an option to the underlying MLWriter.
overwrite()	Overwrites if the output path already exists.
save(path)	Save the ML instance to the input path.
saveImpl(path)	save() handles overwriting and then calls this method.
session(sparkSession)	Sets the Spark Session to use for saving/loading.

Attributes

sc	Returns the underlyingSparkContext.
sparkSession	Returns the user-specified Spark Session or the default.

Methods Documentation

option(key,value)[source]#

Adds an option to the underlying MLWriter. See the documentation for the specific model’swriter for possible options. The option name (key) is case-insensitive.

overwrite()[source]#

Overwrites if the output path already exists.

save(path)[source]#

Save the ML instance to the input path.

saveImpl(path)[source]#

save() handles overwriting and then calls this method. Subclasses should override thismethod to implement the actual saving of the instance.

session(sparkSession)#

Sets the Spark Session to use for saving/loading.

Attributes Documentation

sc#

Returns the underlyingSparkContext.

sparkSession#

Returns the user-specified Spark Session or the default.