Help needed to get probabilities - MyClassification template

[Somnath Yadav]

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*)