MLWriter

class pyspark.ml.util.MLWriter[source]

Utility class that can save ML instances.

New in version 2.0.0.

Methods

Attributes

Methods Documentation

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 this method to implement the actual saving of the instance.

session(sparkSession)

Sets the Spark Session to use for saving/loading.

Attributes Documentation

sc

Returns the underlying SparkContext.

sparkSession

Returns the user-specified Spark Session or the default.