Yep, row of Matrix theta is the number of classes and column of theta is
the number of features.
2016-01-13 10:47 GMT+08:00 Andy Davidson :
> I am trying to debug my trained model by exploring theta
> Theta is a Matrix. The java Doc for Matrix says that it is
Thanks
Andy
From: Yanbo Liang <yblia...@gmail.com>
Date: Wednesday, January 13, 2016 at 6:29 AM
To: Andrew Davidson <a...@santacruzintegration.com>
Cc: "user @spark" <user@spark.apache.org>
Subject: Re: ml.classification.NaiveBayesModel how to reshape theta
I am trying to debug my trained model by exploring theta
Theta is a Matrix. The java Doc for Matrix says that it is column major
formate
I have trained a NaiveBayesModel. Is the number of classes == to the number
of rows?
int numRows = nbModel.numClasses();
int numColumns =