Not directly. If you could access brzPi and brzTheta in the NaiveBayesModel, you could repeat its same computation in predict() and exponentiate it to get back class probabilities, since input and internal values are in log space.
Hm I wonder how people feel about exposing those fields or a different method to expose class probabilities? Seems useful since it is conceptually directly available. On Nov 10, 2014 5:46 AM, "jatinpreet" <jatinpr...@gmail.com> wrote: > Hi, > > Is there a way to get the confidence value of a prediction with MLlib's > implementation of Naive Baye's classification. I wish to eliminate the > samples that were classified with low confidence. > > Thanks, > Jatin > > > > ----- > Novice Big Data Programmer > -- > View this message in context: > http://apache-spark-user-list.1001560.n3.nabble.com/MLlib-Naive-Bayes-classifier-confidence-tp18456.html > Sent from the Apache Spark User List mailing list archive at Nabble.com. > > --------------------------------------------------------------------- > To unsubscribe, e-mail: user-unsubscr...@spark.apache.org > For additional commands, e-mail: user-h...@spark.apache.org > >
