-
Notifications
You must be signed in to change notification settings - Fork 28.9k
[SPARK-9312] [ML] Added max confidence factor to OneVsRestModel #7652
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
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.
Without commenting on the suitability of the PR -- I can at least say that throw new Exception is never a great idea. Use UnsupportedOperationException here, for example.
|
@badriub I just added some thoughts on the JIRA. Could you please take a look and respond there? Thanks! |
|
I have replied on the JIRA, please take a look. Thanks. |
predictive probability since it was not needed.
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.
Used the Intellj importer plugin as recommended on the style guide.
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.
You may need to adjust the import organizer settings, though. The order should be Java, Scala, other, Spark.
|
@jkbradley : I've reverted the changes for LogisticRegression (for entire class hierarchy) since they were not needed. I've updated JIRA and PR as discussed. |
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 will be easier for you if you have OneVsRestParams extend ClassifierParams. That will include the rawPredictionCol, plus the updated validateAndTransformSchema method.
|
That should do it for a first pass. I'll check back for updates. Thanks!
|
|
@jkbradley @badriub Any updates on this issue? From diff I see that I'm happy to help if required to get it through |
|
Is there any news on this branch? we would benefit a lot from this feature. |
|
Can one of the admins verify this patch? |
WRT OVR,
I've extracted the rawProbability as the maxProbability of all classes and added a new column in the returned dataframe.
I've fixed the test to include the new column "rawProbability"