[
https://issues.apache.org/jira/browse/SPARK-14926?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=15556180#comment-15556180
]
Joseph K. Bradley commented on SPARK-14926:
-------------------------------------------
Sorry, just catching up on this. I commented on [[email protected]]'s PR
with what should be workable code. I'll try to check back soon.
[~KrishnaKalyan3] Perhaps another JIRA? Thanks though!
> OneVsRest labelMetadata uses incorrect name
> -------------------------------------------
>
> Key: SPARK-14926
> URL: https://issues.apache.org/jira/browse/SPARK-14926
> Project: Spark
> Issue Type: Bug
> Components: ML
> Affects Versions: 1.4.1, 1.5.2, 1.6.1, 2.0.0
> Reporter: Joseph K. Bradley
> Priority: Trivial
>
> OneVsRestModel applies {{labelMetadata}} to the output column, but the
> metadata could contain the wrong name. The attribute name should be modified
> to match {{predictionCol}}.
> Here is the relevant location:
> [[https://github.com/apache/spark/blob/2a3d39f48b1a7bb462e17e80e243bbc0a94d802e/mllib/src/main/scala/org/apache/spark/ml/classification/OneVsRest.scala#L200]]
--
This message was sent by Atlassian JIRA
(v6.3.4#6332)
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
