Examples with Output org.dmg.pmml.Output used on opensource projects

Example 1 with Output

use of org.dmg.pmml.Output in project jpmml-r by jpmml.

the class BinaryTreeConverter method encodeModel.

@Override
public TreeModel encodeModel(Schema schema) {
    S4Object binaryTree = getObject();
    RGenericVector tree = (RGenericVector) binaryTree.getAttributeValue("tree");
    Output output;
    switch(this.miningFunction) {
        case REGRESSION:
            output = new Output();
            break;
        case CLASSIFICATION:
            CategoricalLabel categoricalLabel = (CategoricalLabel) schema.getLabel();
            output = ModelUtil.createProbabilityOutput(DataType.DOUBLE, categoricalLabel);
            break;
        default:
            throw new IllegalArgumentException();
    }
    output.addOutputFields(ModelUtil.createEntityIdField(FieldName.create("nodeId")));
    TreeModel treeModel = encodeTreeModel(tree, schema).setOutput(output);
    return treeModel;
}

Example 2 with Output

use of org.dmg.pmml.Output in project jpmml-sparkml by jpmml.

the class ModelConverter method registerModel.

public org.dmg.pmml.Model registerModel(SparkMLEncoder encoder) {
    Schema schema = encodeSchema(encoder);
    Label label = schema.getLabel();
    org.dmg.pmml.Model model = encodeModel(schema);
    List<OutputField> sparkOutputFields = registerOutputFields(label, encoder);
    if (sparkOutputFields != null && sparkOutputFields.size() > 0) {
        org.dmg.pmml.Model lastModel = getLastModel(model);
        Output output = lastModel.getOutput();
        if (output == null) {
            output = new Output();
            lastModel.setOutput(output);
        }
        List<OutputField> outputFields = output.getOutputFields();
        outputFields.addAll(0, sparkOutputFields);
    }
    return model;
}