use of org.dmg.pmml.NumericPredictorDocument.NumericPredictor in project knime-core by knime.
the class ClassificationGBTModelExporter method addRegressionTables.
private void addRegressionTables(final RegressionModel regression) {
for (int i = 0; i < getGBTModel().getNrClasses(); i++) {
RegressionTable regTable = regression.addNewRegressionTable();
regTable.setIntercept(0);
regTable.setTargetCategory(getClassLabel(i));
NumericPredictor np = regTable.addNewNumericPredictor();
np.setName(logitName(i));
np.setCoefficient(1d);
}
}
Aggregations