[
https://issues.apache.org/jira/browse/SPARK-14076?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=15210226#comment-15210226
]
Yanbo Liang commented on SPARK-14076:
-------------------------------------
[~mengxr] Because NaiveBayesModel extends from ProbabilisticClassificationModel
and the output prediction column was produced there, so I think we should
modify the transform function of ProbabilisticClassificationModel to add the
attributes. Other ClassificationModels who also extends from
ProbabilisticClassificationModel can benefit from this improvement.
If I understand it correctly, I can work on this.
> Naive Bayes should output attributes in predictions
> ---------------------------------------------------
>
> Key: SPARK-14076
> URL: https://issues.apache.org/jira/browse/SPARK-14076
> Project: Spark
> Issue Type: Improvement
> Components: ML
> Affects Versions: 2.0.0
> Reporter: Xiangrui Meng
> Labels: naive-bayes
>
> In naive Bayes, we outputs predicted labels without feature attributes. So
> from the metadata, we can only tell that it is a nominal feature but the
> values are missing. We should copy the attributes from input label column to
> output prediction column.
--
This message was sent by Atlassian JIRA
(v6.3.4#6332)
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
