You are viewing a plain text version of this content. The canonical link for it is here.

Posted to user@predictionio.apache.org by Somnath Yadav <so...@gmail.com> on 2017/05/09 14:26:54 UTC

Help needed to get probabilities - MyClassification template

Hi,

I am trying to get probabilities using 'predictProbabilities'.  Not sure if
I am missing something.  Below are changes made.  Please suggest.

*Engine.scala*

class PredictedResult(
  val label: *Array[Double]*
) extends Serializable




*NaiveBayesAlgorithm.scala*

  def predict(model: NaiveBayesModel, query: Query): PredictedResult = {
    val label = model.predictProbabilities(Vectors.dense(
      Array(query.attr0, query.attr1, query.attr2)
    ))
    new PredictedResult(*label.toArray*)