diff options
author | Xusen Yin <yinxusen@gmail.com> | 2016-01-26 21:16:56 -0800 |
---|---|---|
committer | Joseph K. Bradley <joseph@databricks.com> | 2016-01-26 21:16:56 -0800 |
commit | 4db255c7aa756daa224d61905db745b6bccc9173 (patch) | |
tree | ab9a7b9184ec9bc7e2a353e74bad45be738296d5 /python | |
parent | bae3c9a4eb0c320999e5dbafd62692c12823e07d (diff) | |
download | spark-4db255c7aa756daa224d61905db745b6bccc9173.tar.gz spark-4db255c7aa756daa224d61905db745b6bccc9173.tar.bz2 spark-4db255c7aa756daa224d61905db745b6bccc9173.zip |
[SPARK-12780] Inconsistency returning value of ML python models' properties
https://issues.apache.org/jira/browse/SPARK-12780
Author: Xusen Yin <yinxusen@gmail.com>
Closes #10724 from yinxusen/SPARK-12780.
Diffstat (limited to 'python')
-rw-r--r-- | python/pyspark/ml/feature.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/python/pyspark/ml/feature.py b/python/pyspark/ml/feature.py index 22081233b0..d017a23188 100644 --- a/python/pyspark/ml/feature.py +++ b/python/pyspark/ml/feature.py @@ -1323,7 +1323,7 @@ class StringIndexer(JavaEstimator, HasInputCol, HasOutputCol, HasHandleInvalid): >>> sorted(set([(i[0], i[1]) for i in td.select(td.id, td.indexed).collect()]), ... key=lambda x: x[0]) [(0, 0.0), (1, 2.0), (2, 1.0), (3, 0.0), (4, 0.0), (5, 1.0)] - >>> inverter = IndexToString(inputCol="indexed", outputCol="label2", labels=model.labels()) + >>> inverter = IndexToString(inputCol="indexed", outputCol="label2", labels=model.labels) >>> itd = inverter.transform(td) >>> sorted(set([(i[0], str(i[1])) for i in itd.select(itd.id, itd.label2).collect()]), ... key=lambda x: x[0]) @@ -1365,13 +1365,14 @@ class StringIndexerModel(JavaModel): .. versionadded:: 1.4.0 """ + @property @since("1.5.0") def labels(self): """ Ordered list of labels, corresponding to indices to be assigned. """ - return self._java_obj.labels + return self._call_java("labels") @inherit_doc |