diff options
| author | Bryan Cutler <cutlerb@gmail.com> | 2016-04-13 14:08:57 -0700 |
|---|---|---|
| committer | Joseph K. Bradley <joseph@databricks.com> | 2016-04-13 14:08:57 -0700 |
| commit | fc3cd2f5090b3ba1cfde0fca3b3ce632d0b2f9c4 (patch) | |
| tree | 6ee38d5b95cb6bc5548c6bb1b8da528aa46ce1a9 /python/pyspark/ml/classification.py | |
| parent | 781df499836e4216939e0febdcd5f89d30645759 (diff) | |
| download | spark-fc3cd2f5090b3ba1cfde0fca3b3ce632d0b2f9c4.tar.gz spark-fc3cd2f5090b3ba1cfde0fca3b3ce632d0b2f9c4.tar.bz2 spark-fc3cd2f5090b3ba1cfde0fca3b3ce632d0b2f9c4.zip | |
[SPARK-14472][PYSPARK][ML] Cleanup ML JavaWrapper and related class hierarchy
Currently, JavaWrapper is only a wrapper class for pipeline classes that have Params and JavaCallable is a separate mixin that provides methods to make Java calls. This change simplifies the class structure and to define the Java wrapper in a plain base class along with methods to make Java calls. Also, renames Java wrapper classes to better reflect their purpose.
Ran existing Python ml tests and generated documentation to test this change.
Author: Bryan Cutler <cutlerb@gmail.com>
Closes #12304 from BryanCutler/pyspark-cleanup-JavaWrapper-SPARK-14472.
Diffstat (limited to 'python/pyspark/ml/classification.py')
| -rw-r--r-- | python/pyspark/ml/classification.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/python/pyspark/ml/classification.py b/python/pyspark/ml/classification.py index e64c7a392b..922f8069fa 100644 --- a/python/pyspark/ml/classification.py +++ b/python/pyspark/ml/classification.py @@ -19,7 +19,7 @@ import warnings from pyspark import since from pyspark.ml.util import * -from pyspark.ml.wrapper import JavaEstimator, JavaModel, JavaCallable +from pyspark.ml.wrapper import JavaEstimator, JavaModel, JavaWrapper from pyspark.ml.param import TypeConverters from pyspark.ml.param.shared import * from pyspark.ml.regression import ( @@ -272,7 +272,7 @@ class LogisticRegressionModel(JavaModel, JavaMLWritable, JavaMLReadable): return BinaryLogisticRegressionSummary(java_blr_summary) -class LogisticRegressionSummary(JavaCallable): +class LogisticRegressionSummary(JavaWrapper): """ Abstraction for Logistic Regression Results for a given model. |