-
Notifications
You must be signed in to change notification settings - Fork 28.9k
[SPARK-11892][ML] Model export/import for spark.ml: OneVsRest #9934
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 #46604 has finished for PR 9934 at commit
|
|
Updated for 2.0 cc @jkbradley |
|
Test build #49131 has finished for PR 9934 at commit
|
|
I'll take a look now |
|
Test build #2671 has finished for PR 9934 at commit
|
|
I'll update this after #9971 merged. |
|
Refactoring this one now. |
|
Test build #54411 has finished for PR 9934 at commit
|
|
@jkbradley Ready to review. |
|
Github says this PR does not conflict with master, but when I rebase on master, it has conflicts. Does this happen to you? |
|
Not happen on me. Maybe you can hold off first, I can check it later today because I am off my PC now, then ping you if it passes. |
|
Sure, I'll go ahead and review though. Btw, can you add the "[ML]" tag to the PR title? |
|
Sure Sent from my iPhone
|
|
@jkbradley I merged with master, it seems good. |
|
Test build #54488 has finished for PR 9934 at commit
|
| case stage: MLWritable => // good | ||
| case other => | ||
| throw new UnsupportedOperationException("OneVsRest write will fail " + | ||
| s" because it contains $name which does not implement Writable." + |
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.
"Writable" --> "MLWritable"
| @Since("1.4.0") override val uid: String, | ||
| @Since("1.4.0") labelMetadata: Metadata, | ||
| @Since("1.4.0") override val uid: String, | ||
| @Since("1.4.0") val labelMetadata: Metadata, |
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.
This should be private[ml]. The Since tag was a mistake and can be removed since it does not show up in the public API.
|
I think that should be it. |
|
Test build #54570 has finished for PR 9934 at commit
|
|
@jkbradley All fixed. |
|
LGTM |
|
Thanks! |
|
I made a JIRA for the Python API: https://issues.apache.org/jira/browse/SPARK-14306 |
|
Sure I can take it. Sent from my iPhone
|
|
Thanks! |
|
@jkbradley It seems that OneVsRest is still missing for PySpark. I find the closed PR: #6403. I can help re-submitting it with the PySpark one. |
|
@yinxusen Thanks! That would be great. |
What changes were proposed in this pull request?
https://issues.apache.org/jira/browse/SPARK-11892
Add save/load for spark ml.OneVsRest and its model. Also add OneVsRest and OneVsRestModel in MetaAlgorithmReadWrite.
How was this patch tested?
Test with Scala unit test.