Examples with Node org.dmg.pmml.NodeDocument.Node used on opensource projects

Example 6 with Node

use of org.dmg.pmml.NodeDocument.Node in project knime-core by knime.

the class AbstractRegressionContentParser method createNode.

@Override
public final TreeNodeRegression createNode(final Node node, final TargetColumnHelper<TreeTargetNumericColumnMetaData> targetHelper, final TreeNodeSignature signature, final List<TreeNodeRegression> children) {
    double mean = Double.parseDouble(node.getScore());
    OptionalDouble totalSum = node.getExtensionList().stream().filter(e -> e.getName().equals(TranslationUtil.TOTAL_SUM_KEY)).mapToDouble(e -> Double.parseDouble(e.getValue())).findFirst();
    OptionalDouble sumSquaredDeviation = node.getExtensionList().stream().filter(e -> e.getName().equals(TranslationUtil.SUM_SQUARED_DEVIATION_KEY)).mapToDouble(e -> Double.parseDouble(e.getValue())).findFirst();
    return createNodeInternal(node, targetHelper.getMetaData(), signature, mean, totalSum.orElse(-1), sumSquaredDeviation.orElse(-1), children.toArray(new TreeNodeRegression[children.size()]));
}

Example 7 with Node

use of org.dmg.pmml.NodeDocument.Node in project knime-core by knime.

the class AbstractTreeModelExporter method writeModelToPMML.

/**
 * Writes the KNIME tree model to a PMML tree model.
 * @param treeModel the PMML node which should be filled with the KNIME model
 * @param spec the {@link PMMLPortObjectSpec} of the full PMML document
 * @return the {@link SchemaType} of the tree model
 */
public SchemaType writeModelToPMML(final TreeModel treeModel, final PMMLPortObjectSpec spec) {
    checkColumnTypes(spec);
    PMMLMiningSchemaTranslator.writeMiningSchema(reduceSpec(spec), treeModel);
    treeModel.setModelName("DecisionTree");
    treeModel.setFunctionName(getMiningFunction());
    T rootNode = m_treeModel.getRootNode();
    // set up splitCharacteristic
    if (isMultiSplitRecursive(rootNode)) {
        treeModel.setSplitCharacteristic(SplitCharacteristic.MULTI_SPLIT);
    } else {
        treeModel.setSplitCharacteristic(SplitCharacteristic.BINARY_SPLIT);
    }
    // ----------------------------------------------
    // set up missing value strategy
    treeModel.setMissingValueStrategy(MISSINGVALUESTRATEGY.NONE);
    // -------------------------------------------------
    // set up no true child strategy
    treeModel.setNoTrueChildStrategy(NOTRUECHILDSTRATEGY.RETURN_LAST_PREDICTION);
    // --------------------------------------------------
    // set up tree node
    NodeDocument.Node rootPMMLNode = treeModel.addNewNode();
    addTreeNode(rootPMMLNode, rootNode);
    return TreeModelDocument.TreeModel.type;
}

Example 8 with Node

use of org.dmg.pmml.NodeDocument.Node in project knime-core by knime.

the class TreeModelImporter method importFromPMML.

/**
 * Imports an {@link AbstractTreeModel} from PMML.
 *
 * @param treeModel PMML tree model to import
 * @return a {@link AbstractTreeModel} initialized from <b>treeModel</b>
 */
public M importFromPMML(final TreeModel treeModel) {
    Node rootNode = treeModel.getNode();
    N root = createNodeFromPMML(rootNode, m_signatureFactory.getRootSignature());
    return m_treeFactory.createTree(root);
}

Example 9 with Node

use of org.dmg.pmml.NodeDocument.Node in project knime-core by knime.

the class TreeModelImporter method createNodeFromPMML.

private N createNodeFromPMML(final Node pmmlNode, final TreeNodeSignature signature) {
    List<N> children = new ArrayList<>();
    byte i = 0;
    for (Node child : pmmlNode.getNodeList()) {
        TreeNodeSignature childSignature = m_signatureFactory.getChildSignatureFor(signature, i);
        i++;
        children.add(createNodeFromPMML(child, childSignature));
    }
    TreeNodeCondition condition = m_conditionParser.parseCondition(pmmlNode);
    N node = m_contentParser.createNode(pmmlNode, m_metaDataMapper.getTargetColumnHelper(), signature, children);
    node.setTreeNodeCondition(condition);
    return node;
}