-
Notifications
You must be signed in to change notification settings - Fork 28.6k
[SPARK-27007][PYTHON]add rawPrediction to OneVsRest in PySpark #23910
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.
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
Conversation
Test build #102831 has finished for PR 23910 at commit
|
Test build #102836 has finished for PR 23910 at commit
|
Test build #102838 has finished for PR 23910 at commit
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looking good, with one question
python/pyspark/sql/dataframe.py
Outdated
@@ -1980,6 +1980,29 @@ def withColumn(self, colName, col): | |||
assert isinstance(col, Column), "col should be Column" | |||
return DataFrame(self._jdf.withColumn(colName, col._jc), self.sql_ctx) | |||
|
|||
@ignore_unicode_prefix | |||
def withColumns(self, colNames, cols): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
It looks like the Scala impl has this method, but it's not exposed. Instead of adding it here, can you just use two calls to withColumn
above?
Test build #102924 has finished for PR 23910 at commit
|
Merged to master |
What changes were proposed in this pull request?
Add RawPrediction to OneVsRest in PySpark to make it consistent with scala implementation
How was this patch tested?
Add doctest