/** * Fit the model * * @param data the data to train on */ @Override public void fit(DataSet data) { fit(data.getFeatureMatrix(), data.getLabels()); }
/** * 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()); }