Github user yanboliang commented on the pull request:
https://github.com/apache/spark/pull/7284#issuecomment-121824128
@jkbradley I agreed and have removed "labels" from ml.NaiveBayesModel. At
present the pi and theta use the 0-based indices in ml scope. I have also add
label check
[here](https://github.com/apache/spark/pull/7284/files#diff-f65e90b33459f2f7fd78addb12cc5d1dR95)
to verify the labels of original dataset are 0-based indices rather than
arbitrary Double labels which will lead to erroneous prediction.
As for setParent issue, I also agreed with you and changed
[here](https://github.com/apache/spark/pull/7284/files#diff-f65e90b33459f2f7fd78addb12cc5d1dR174).
As far as I know, almost all the models and transformers are not correctly
set, I can solve it in a follow up PR.
---
If your project is set up for it, you can reply to this email and have your
reply appear on GitHub as well. If your project does not have this feature
enabled and wishes so, or if the feature is enabled but not working, please
contact infrastructure at [email protected] or file a JIRA ticket
with INFRA.
---
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
