aboutsummaryrefslogtreecommitdiff
path: root/python
diff options
context:
space:
mode:
authorXusen Yin <yinxusen@gmail.com>2016-04-18 13:31:48 -0700
committerXiangrui Meng <meng@databricks.com>2016-04-18 13:31:48 -0700
commitf31a62d1b24aea8ddfa40b60378ce065518786e4 (patch)
tree8461f45dcb9a4a2b4fb8a28515903ff92a2d6067 /python
parent28ee15702d9efd52a26a065c6e544b5345a8f65d (diff)
downloadspark-f31a62d1b24aea8ddfa40b60378ce065518786e4.tar.gz
spark-f31a62d1b24aea8ddfa40b60378ce065518786e4.tar.bz2
spark-f31a62d1b24aea8ddfa40b60378ce065518786e4.zip
[SPARK-14440][PYSPARK] Remove pipeline specific reader and writer
## What changes were proposed in this pull request? https://issues.apache.org/jira/browse/SPARK-14440 Remove * PipelineMLWriter * PipelineMLReader * PipelineModelMLWriter * PipelineModelMLReader and modify comments. ## How was this patch tested? test with unit test. Author: Xusen Yin <yinxusen@gmail.com> Closes #12216 from yinxusen/SPARK-14440.
Diffstat (limited to 'python')
-rw-r--r--python/pyspark/ml/pipeline.py53
1 files changed, 7 insertions, 46 deletions
diff --git a/python/pyspark/ml/pipeline.py b/python/pyspark/ml/pipeline.py
index 6f599b5159..e2651aebdf 100644
--- a/python/pyspark/ml/pipeline.py
+++ b/python/pyspark/ml/pipeline.py
@@ -30,24 +30,6 @@ from pyspark.mllib.common import inherit_doc
@inherit_doc
-class PipelineMLWriter(JavaMLWriter):
- """
- Private Pipeline utility class that can save ML instances through their Scala implementation.
-
- We can currently use JavaMLWriter, rather than MLWriter, since Pipeline implements _to_java.
- """
-
-
-@inherit_doc
-class PipelineMLReader(JavaMLReader):
- """
- Private utility class that can load Pipeline instances through their Scala implementation.
-
- We can currently use JavaMLReader, rather than MLReader, since Pipeline implements _from_java.
- """
-
-
-@inherit_doc
class Pipeline(Estimator, MLReadable, MLWritable):
"""
A simple pipeline, which acts as an estimator. A Pipeline consists
@@ -154,8 +136,8 @@ class Pipeline(Estimator, MLReadable, MLWritable):
@since("2.0.0")
def write(self):
- """Returns an JavaMLWriter instance for this ML instance."""
- return PipelineMLWriter(self)
+ """Returns an MLWriter instance for this ML instance."""
+ return JavaMLWriter(self)
@since("2.0.0")
def save(self, path):
@@ -166,7 +148,7 @@ class Pipeline(Estimator, MLReadable, MLWritable):
@since("2.0.0")
def read(cls):
"""Returns an MLReader instance for this class."""
- return PipelineMLReader(cls)
+ return JavaMLReader(cls)
@classmethod
def _from_java(cls, java_stage):
@@ -202,27 +184,6 @@ class Pipeline(Estimator, MLReadable, MLWritable):
@inherit_doc
-class PipelineModelMLWriter(JavaMLWriter):
- """
- Private PipelineModel utility class that can save ML instances through their Scala
- implementation.
-
- We can (currently) use JavaMLWriter, rather than MLWriter, since PipelineModel implements
- _to_java.
- """
-
-
-@inherit_doc
-class PipelineModelMLReader(JavaMLReader):
- """
- Private utility class that can load PipelineModel instances through their Scala implementation.
-
- We can currently use JavaMLReader, rather than MLReader, since PipelineModel implements
- _from_java.
- """
-
-
-@inherit_doc
class PipelineModel(Model, MLReadable, MLWritable):
"""
Represents a compiled pipeline with transformers and fitted models.
@@ -254,8 +215,8 @@ class PipelineModel(Model, MLReadable, MLWritable):
@since("2.0.0")
def write(self):
- """Returns an JavaMLWriter instance for this ML instance."""
- return PipelineModelMLWriter(self)
+ """Returns an MLWriter instance for this ML instance."""
+ return JavaMLWriter(self)
@since("2.0.0")
def save(self, path):
@@ -265,8 +226,8 @@ class PipelineModel(Model, MLReadable, MLWritable):
@classmethod
@since("2.0.0")
def read(cls):
- """Returns an JavaMLReader instance for this class."""
- return PipelineModelMLReader(cls)
+ """Returns an MLReader instance for this class."""
+ return JavaMLReader(cls)
@classmethod
def _from_java(cls, java_stage):