-
Notifications
You must be signed in to change notification settings - Fork 28.4k
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
[SPARK-14926] [ML] OneVsRest labelMetadata uses incorrect name #13762
[SPARK-14926] [ML] OneVsRest labelMetadata uses incorrect name #13762
Conversation
@josh-howes Did you try compiling the code? |
Ping |
Test build #3251 has finished for PR 13762 at commit
|
bd510ec
to
b4badae
Compare
@SparkQA build |
I think we should close this PR. |
I recently pushed a change to this but @SparkQA didn't kick off an automated build. Is that because I squashed my commits? |
Jenkins test this please |
Test build #65256 has finished for PR 13762 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.
Also, could you please add a line to a unit test to verify that the predictionCol has the correct metadata? Thank you!
@@ -196,8 +196,13 @@ final class OneVsRestModel private[ml] ( | |||
} | |||
|
|||
// output label and label metadata as prediction | |||
val predictionMetadata = new MetadataBuilder() | |||
.withMetadata(labelMetadata) | |||
.putString("name", predictionCol.name) |
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 go through the standard ML attribute constructors to modify labelMetadata:
val predictionMetadata = Attribute.fromMetadata(labelMetadata)
.withName($(predictionCol))
.toMetadata()
Test build #3299 has finished for PR 13762 at commit
|
Hi @josh-howes is it still active? If so, please address the comments above. Otherwise, I would rather like to propose to close this. |
This contribution is my original work and I license the work to the project under the project's open source license.
What changes were proposed in this pull request?
OneVsRestModel applies labelMetadata to the output column, but the metadata could contain the wrong name. The attribute name was modified to match predictionCol.
How was this patch tested?
Manual Tests