0
Я создал NaiveBayesModel в MLlib. Он отлично работает, но я хотел бы вернуть вероятности каждого класса для заданного ввода вместо окончательного и единственного решения модели, то есть, принадлежит ли вход классу 1.0 или классу 0.0. Есть идеи?Как получить вероятности (вместо одного - фактического класса) от NaiveBayesModel в MLlib?
Спасибо за ваш ответ. Это верно. Spark 1.5 будет поддерживать это на Naive Bayes. Моя проблема, однако, в том, что я использую Spark 1.1.0, и я не могу обновить, поэтому мне нужно найти обходной путь. Есть идеи? http://stackoverflow.com/questions/31842502/how-to-get-the-probabilities-of-classes-in-spark-naive-bayes-classifier – user706838
Это то, что необходимо сохранить при вычислении модели, с доступ к коду можно было сделать точно. Если бы я был вами, я бы проверил некоторую реализацию Naive Bayes, которая имеет такую реализацию, как пример, а затем проверит здесь, как она реализована, а затем внесите изменения, чтобы получить ее для вывода. https://github.com/apache/spark/blob/master/mllib/src/main/scala/org/apache/spark/mllib/classification/NaiveBayes.scala –