Search in sources :

Example 41 with Pair

use of org.knime.core.util.Pair in project knime-core by knime.

the class TypeSelectionConfiguration method configure.

@Override
public List<Pair<String, PaneConfigurationDialog>> configure(final DataTableSpec spec) {
    List<Pair<String, PaneConfigurationDialog>> r = new ArrayList<Pair<String, PaneConfigurationDialog>>();
    for (DropPaneConfig dpc : m_panelList.values()) {
        String type = dpc.getSelection().get(0);
        DataType t = StringCell.TYPE;
        if (type.equals("IntCell")) {
            t = IntCell.TYPE;
        } else if (type.equals("DoubleCell")) {
            t = DoubleCell.TYPE;
        } else if (type.equals("LongCell")) {
            t = LongCell.TYPE;
        } else if (type.equals("DateAndTimeCell")) {
            t = DateAndTimeCell.TYPE;
        } else if (type.equals("XMLCell")) {
            t = XMLCell.TYPE;
        } else if (type.equals("PMMLCell")) {
            t = PMMLCell.TYPE;
        }
        for (DataColumnSpec dcs : spec) {
            if (dcs.getType().equals(t)) {
                r.add(new Pair<String, PaneConfigurationDialog>(dcs.getName(), dpc.getDialog()));
            }
        }
    }
    return r;
}
Also used : DataColumnSpec(org.knime.core.data.DataColumnSpec) ArrayList(java.util.ArrayList) DataType(org.knime.core.data.DataType) DropPaneConfig(org.knime.base.node.preproc.draganddroppanel.droppanes.DropPaneConfig) Pair(org.knime.core.util.Pair)

Example 42 with Pair

use of org.knime.core.util.Pair in project knime-core by knime.

the class AbstractGBTModelImporter method readTreeModel.

private Pair<TreeModelRegression, Map<TreeNodeSignature, Double>> readTreeModel(final Segment segment) {
    GBTRegressionContentParser contentParser = new GBTRegressionContentParser();
    TreeModelImporter<TreeNodeRegression, TreeModelRegression, TreeTargetNumericColumnMetaData> treeImporter = new TreeModelImporter<TreeNodeRegression, TreeModelRegression, TreeTargetNumericColumnMetaData>(m_metaDataMapper, m_conditionParser, m_signatureFactory, contentParser, m_treeFactory);
    TreeModel treeModel = segment.getTreeModel();
    TreeModelRegression tree = treeImporter.importFromPMML(treeModel);
    Map<TreeNodeSignature, Double> coefficientMap = contentParser.getCoefficientMap();
    return new Pair<>(tree, coefficientMap);
}
Also used : TreeModel(org.dmg.pmml.TreeModelDocument.TreeModel) TreeTargetNumericColumnMetaData(org.knime.base.node.mine.treeensemble2.data.TreeTargetNumericColumnMetaData) TreeNodeSignature(org.knime.base.node.mine.treeensemble2.model.TreeNodeSignature) TreeNodeRegression(org.knime.base.node.mine.treeensemble2.model.TreeNodeRegression) TreeModelRegression(org.knime.base.node.mine.treeensemble2.model.TreeModelRegression) Pair(org.knime.core.util.Pair)

Example 43 with Pair

use of org.knime.core.util.Pair in project knime-core by knime.

the class AbstractGBTModelImporter method readSumSegmentation.

protected Pair<List<TreeModelRegression>, List<Map<TreeNodeSignature, Double>>> readSumSegmentation(final Segmentation segmentation) {
    List<Segment> segments = segmentation.getSegmentList();
    List<TreeModelRegression> trees = new ArrayList<>(segments.size());
    List<Map<TreeNodeSignature, Double>> coefficientMaps = new ArrayList<>(segments.size());
    for (Segment segment : segments) {
        Pair<TreeModelRegression, Map<TreeNodeSignature, Double>> treeCoeffientMapPair = readTreeModel(segment);
        trees.add(treeCoeffientMapPair.getFirst());
        coefficientMaps.add(treeCoeffientMapPair.getSecond());
    }
    return new Pair<>(trees, coefficientMaps);
}
Also used : ArrayList(java.util.ArrayList) Map(java.util.Map) Segment(org.dmg.pmml.SegmentDocument.Segment) TreeModelRegression(org.knime.base.node.mine.treeensemble2.model.TreeModelRegression) Pair(org.knime.core.util.Pair)

Example 44 with Pair

use of org.knime.core.util.Pair in project knime-core by knime.

the class AppendVariableToTableNodeModel method getAllVariables.

private List<Pair<String, FlowVariable.Type>> getAllVariables() {
    Map<String, FlowVariable> currentVars = getAvailableFlowVariables();
    List<Pair<String, FlowVariable.Type>> variables;
    variables = new ArrayList<Pair<String, FlowVariable.Type>>();
    for (FlowVariable v : currentVars.values()) {
        variables.add(new Pair<String, FlowVariable.Type>(v.getName(), v.getType()));
    }
    return variables;
}
Also used : PortType(org.knime.core.node.port.PortType) DataType(org.knime.core.data.DataType) FlowVariable(org.knime.core.node.workflow.FlowVariable) Pair(org.knime.core.util.Pair)

Example 45 with Pair

use of org.knime.core.util.Pair in project knime-core by knime.

the class VariableToTableNodeModel method getAllVariables.

private List<Pair<String, FlowVariable.Type>> getAllVariables() {
    Map<String, FlowVariable> currentVars = getAvailableFlowVariables();
    List<Pair<String, FlowVariable.Type>> variables;
    variables = new ArrayList<Pair<String, FlowVariable.Type>>();
    for (FlowVariable v : currentVars.values()) {
        variables.add(new Pair<String, FlowVariable.Type>(v.getName(), v.getType()));
    }
    return variables;
}
Also used : PortType(org.knime.core.node.port.PortType) DataType(org.knime.core.data.DataType) FlowVariable(org.knime.core.node.workflow.FlowVariable) Pair(org.knime.core.util.Pair)

Aggregations

Pair (org.knime.core.util.Pair)54 ArrayList (java.util.ArrayList)17 DataCell (org.knime.core.data.DataCell)14 DataType (org.knime.core.data.DataType)13 InvalidSettingsException (org.knime.core.node.InvalidSettingsException)13 PortType (org.knime.core.node.port.PortType)13 LinkedHashMap (java.util.LinkedHashMap)11 Map (java.util.Map)10 DataColumnSpec (org.knime.core.data.DataColumnSpec)10 HashMap (java.util.HashMap)9 HashSet (java.util.HashSet)9 DataTableSpec (org.knime.core.data.DataTableSpec)9 FlowVariable (org.knime.core.node.workflow.FlowVariable)9 DataRow (org.knime.core.data.DataRow)8 StringCell (org.knime.core.data.def.StringCell)7 SettingsModelString (org.knime.core.node.defaultnodesettings.SettingsModelString)7 DataColumnSpecCreator (org.knime.core.data.DataColumnSpecCreator)6 ColumnRearranger (org.knime.core.data.container.ColumnRearranger)6 DefaultRow (org.knime.core.data.def.DefaultRow)6 DoubleCell (org.knime.core.data.def.DoubleCell)6