aboutsummaryrefslogtreecommitdiff
path: root/python
diff options
context:
space:
mode:
Diffstat (limited to 'python')
-rw-r--r--python/pyspark/ml/classification.py13
-rw-r--r--python/pyspark/ml/regression.py12
2 files changed, 25 insertions, 0 deletions
diff --git a/python/pyspark/ml/classification.py b/python/pyspark/ml/classification.py
index 4cbe7fbd48..2e468f67b8 100644
--- a/python/pyspark/ml/classification.py
+++ b/python/pyspark/ml/classification.py
@@ -15,6 +15,9 @@
# limitations under the License.
#
+import warnings
+
+from pyspark import since
from pyspark.ml.util import keyword_only
from pyspark.ml.wrapper import JavaEstimator, JavaModel
from pyspark.ml.param.shared import *
@@ -189,9 +192,19 @@ class LogisticRegressionModel(JavaModel):
"""
Model weights.
"""
+
+ warnings.warn("weights is deprecated. Use coefficients instead.")
return self._call_java("weights")
@property
+ @since("1.6.0")
+ def coefficients(self):
+ """
+ Model coefficients.
+ """
+ return self._call_java("coefficients")
+
+ @property
def intercept(self):
"""
Model intercept.
diff --git a/python/pyspark/ml/regression.py b/python/pyspark/ml/regression.py
index dc68815556..ab26616f4a 100644
--- a/python/pyspark/ml/regression.py
+++ b/python/pyspark/ml/regression.py
@@ -15,6 +15,8 @@
# limitations under the License.
#
+import warnings
+
from pyspark import since
from pyspark.ml.util import keyword_only
from pyspark.ml.wrapper import JavaEstimator, JavaModel
@@ -117,9 +119,19 @@ class LinearRegressionModel(JavaModel):
"""
Model weights.
"""
+
+ warnings.warn("weights is deprecated. Use coefficients instead.")
return self._call_java("weights")
@property
+ @since("1.6.0")
+ def coefficients(self):
+ """
+ Model coefficients.
+ """
+ return self._call_java("coefficients")
+
+ @property
@since("1.4.0")
def intercept(self):
"""