spark-user mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From Sean Owen <so...@cloudera.com>
Subject Re: How to compute the probability of each class in Naive Bayes
Date Tue, 01 Sep 2015 09:57:39 GMT
(pedantic: it's the log-probabilities)

On Tue, Sep 1, 2015 at 10:48 AM, Yanbo Liang <ybliang8@gmail.com> wrote:
> Actually
> brzPi + brzTheta * testData.toBreeze
> is the probabilities of the input Vector on each class, however it's a
> Breeze Vector.
> Pay attention the index of this Vector need to map to the corresponding
> label index.
>
> 2015-08-28 20:38 GMT+08:00 Adamantios Corais <adamantios.corais@gmail.com>:
>>
>> Hi,
>>
>> I am trying to change the following code so as to get the probabilities of
>> the input Vector on each class (instead of the class itself with the highest
>> probability). I know that this is already available as part of the most
>> recent release of Spark but I have to use Spark 1.1.0.
>>
>> Any help is appreciated.
>>
>>> override def predict(testData: Vector): Double = {
>>>     labels(brzArgmax(brzPi + brzTheta * testData.toBreeze))
>>>   }
>>
>>
>>>
>>> https://github.com/apache/spark/blob/v1.1.0/mllib/src/main/scala/org/apache/spark/mllib/classification/NaiveBayes.scala
>>
>>
>> // Adamantios
>>
>>
>

---------------------------------------------------------------------
To unsubscribe, e-mail: user-unsubscribe@spark.apache.org
For additional commands, e-mail: user-help@spark.apache.org


Mime
View raw message