Exemplo n.º 1
0
 /**
  * Fit the model
  *
  * @param data the data to train on
  */
 @Override
 public void fit(DataSet data) {
   fit(data.getFeatureMatrix(), data.getLabels());
 }
Exemplo n.º 2
0
 /**
  * Sets the input and labels and returns a score for the prediction wrt true labels
  *
  * @param data the data to score
  * @return the score for the given input,label pairs
  */
 public double f1Score(DataSet data) {
   return f1Score(data.getFeatureMatrix(), data.getLabels());
 }