Skip to content

[SPARK-12780] Inconsistency returning value of ML python models' properties #10724

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wants to merge 6 commits into from
Closed

[SPARK-12780] Inconsistency returning value of ML python models' properties #10724

Changes from all commits
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
993dc9c
change labels
yinxusen Jan 12, 2016
c5a170d
change Array to JavaList when encountering Pickle
yinxusen Jan 13, 2016
becc49c
add comment
yinxusen Jan 13, 2016
f5623b6
remove code in Scala
yinxusen Jan 15, 2016
293efba
change to call java
yinxusen Jan 26, 2016
c0cf68c
Merge branch 'master' into SPARK-12780
yinxusen Jan 26, 2016
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 3 additions & 2 deletions python/pyspark/ml/feature.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1387,7 +1387,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])
Expand Down Expand Up @@ -1429,13 +1429,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
Expand Down