-
Notifications
You must be signed in to change notification settings - Fork 29k
[SPARK-6791] [ML] Add read/write for CrossValidator and Evaluators #9848
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 #46369 has finished for PR 9848 at commit
|
|
@mengxr @yanboliang Ready for review! |
|
Test build #46394 has finished for PR 9848 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.
Minor: organize imports
|
LGTM for me. |
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.
Shall we add checks for OneVSRest and other meta-algorithms? It is okay to create a JIRA and leave a TODO here.
|
Made one pass. |
…her metadata. Cleanups to CrossValidator save/load based on code review. Refactoring to eliminate some duplicate code.
|
OK, I believe that addressed everything. This includes a little refactoring to make sure callers to the default metadata save/load methods are using them correctly (and it eliminates some duplicate code). |
|
Test build #46464 has finished for PR 9848 at commit
|
I believe this works for general estimators within CrossValidator, including compound estimators. (See the complex unit test.) Added read/write for all 3 Evaluators as well. CC: mengxr yanboliang Author: Joseph K. Bradley <joseph@databricks.com> Closes #9848 from jkbradley/cv-io. (cherry picked from commit a6fda0b) Signed-off-by: Xiangrui Meng <meng@databricks.com>
|
LGTM. Merged into master and branch-1.6. Thanks! |
I believe this works for general estimators within CrossValidator, including compound estimators. (See the complex unit test.)
Added read/write for all 3 Evaluators as well.
CC: @mengxr @yanboliang