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 <a...@santacruzintegration.com>: > 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 = nbModel.numFeatures(); > > > Kind regards > > > Andy >
