Я построил NaiveBayesModel в MLlib. Он отлично работает, но я хотел бы вернуть вероятности каждого класса для заданного ввода вместо окончательного и единственного решения модели, то есть, относится ли ввод к классу 1.0 или классу 0.0. Любые идеи?
Как получить вероятности (вместо одного фактического класса) из NaiveBayesModel в MLlib?
Ответы (1)
Я не думаю, что они еще поддерживают это, но я знаю, что в их дорожной карте есть добавление вероятностей к результатам предсказания нескольких классификаторов.
person
Dr VComas
schedule
12.08.2015
Спасибо за Ваш ответ. Вот так. Spark 1.5 будет поддерживать это на Наивном Байесе. Однако моя проблема в том, что я использую Spark 1.1.0 и не могу обновить его, поэтому мне нужно найти обходной путь. Любые идеи? stackoverflow.com/questions/31842502/
- person user706838; 12.08.2015
Это то, что нужно сохранить при расчете модели, имея доступ к коду, это можно было бы сделать наверняка. Если бы я был на вашем месте, я бы проверил некоторую реализацию наивного байесовского метода, в которой есть эта реализация, просто в качестве примера, а затем проверил здесь, как она реализована, а затем внес изменения, чтобы заставить ее выводить это. github.com/apache/spark/blob/master/mllib/src/main/scala/org/
- person Dr VComas; 12.08.2015