Examples with MiningField org.dmg.pmml.MiningFieldDocument.MiningField used on opensource projects

Example 6 with MiningField

use of org.dmg.pmml.MiningFieldDocument.MiningField in project knime-core by knime.

the class ClassificationGBTModelExporter method addAggregationMiningScheme.

private void addAggregationMiningScheme(final RegressionModel regression) {
    MiningSchema miningSchema = regression.addNewMiningSchema();
    // add target field
    MiningField targetField = miningSchema.addNewMiningField();
    targetField.setName(getGBTModel().getMetaData().getTargetMetaData().getAttributeName());
    targetField.setUsageType(FIELDUSAGETYPE.TARGET);
    // add class logits
    for (int i = 0; i < getGBTModel().getNrClasses(); i++) {
        MiningField logit = miningSchema.addNewMiningField();
        logit.setName(logitName(i));
    }
}