use of org.dmg.pmml.DerivedFieldDocument.DerivedField in project knime-core by knime.
the class PMMLNeuralNetworkTranslator method addInputLayer.
/**
* Writes the PMML input layer of the MLP.
*
* @param nnModel
* the Neural Network model.
* @param mlp
* the underlying {@link MultiLayerPerceptron}.
*/
protected void addInputLayer(final NeuralNetwork nnModel, final MultiLayerPerceptron mlp) {
Layer inputlayer = mlp.getLayer(0);
Perceptron[] inputperceptrons = inputlayer.getPerceptrons();
HashMap<String, Integer> inputmap = mlp.getInputMapping();
NeuralInputs neuralInputs = nnModel.addNewNeuralInputs();
neuralInputs.setNumberOfInputs(BigInteger.valueOf(inputperceptrons.length));
for (int i = 0; i < inputperceptrons.length; i++) {
NeuralInput neuralInput = neuralInputs.addNewNeuralInput();
neuralInput.setId(0 + "," + i);
// search corresponding input column
String colname = "";
for (Entry<String, Integer> e : inputmap.entrySet()) {
if (e.getValue().equals(i)) {
colname = e.getKey();
}
}
DerivedField df = neuralInput.addNewDerivedField();
df.setOptype(OPTYPE.CONTINUOUS);
df.setDataType(DATATYPE.DOUBLE);
FieldRef fieldRef = df.addNewFieldRef();
fieldRef.setField(m_nameMapper.getDerivedFieldName(colname));
}
}
use of org.dmg.pmml.DerivedFieldDocument.DerivedField in project knime-core by knime.
the class PMMLOne2ManyTranslator method createDerivedFields.
private DerivedField[] createDerivedFields() {
List<DerivedField> derivedFields = new ArrayList<DerivedField>();
for (Map.Entry<String, List<Pair<String, String>>> entry : m_columnMapping.entrySet()) {
String columnName = entry.getKey();
String derivedName = m_mapper.getDerivedFieldName(columnName);
for (Pair<String, String> nameValue : entry.getValue()) {
DerivedField derivedField = DerivedField.Factory.newInstance();
derivedField.setName(nameValue.getFirst());
derivedField.setOptype(OPTYPE.ORDINAL);
derivedField.setDataType(DATATYPE.INTEGER);
NormDiscrete normDiscrete = derivedField.addNewNormDiscrete();
normDiscrete.setField(derivedName);
normDiscrete.setValue(nameValue.getSecond());
normDiscrete.setMapMissingTo(0);
derivedFields.add(derivedField);
}
}
return derivedFields.toArray(new DerivedField[0]);
}
use of org.dmg.pmml.DerivedFieldDocument.DerivedField in project knime-core by knime.
the class PMMLMany2OneTranslator method createDerivedField.
private DerivedField createDerivedField() {
final DerivedField derivedField = DerivedField.Factory.newInstance();
derivedField.setName(m_appendedCol);
derivedField.setDataType(DATATYPE.STRING);
derivedField.setOptype(OPTYPE.CATEGORICAL);
Apply parentApply = null;
for (String col : m_sourceCols) {
Apply ifApply;
if (parentApply == null) {
ifApply = derivedField.addNewApply();
} else {
ifApply = parentApply.addNewApply();
}
ifApply.setFunction("if");
Apply innerIf = ifApply.addNewApply();
innerIf.setFunction("equal");
innerIf.addNewFieldRef().setField(col);
if (m_method == IncludeMethod.Maximum || m_method == IncludeMethod.Minimum) {
Apply a = innerIf.addNewApply();
a.setFunction(IncludeMethod.Maximum == m_method ? "max" : "min");
for (String s : m_sourceCols) {
a.addNewFieldRef().setField(s);
}
} else {
// if (m_method == IncludeMethod.Binary) {
innerIf.addNewConstant().setStringValue("1");
}
ifApply.addNewConstant().setStringValue(col);
parentApply = ifApply;
}
if (parentApply != null) {
parentApply.addNewConstant().setStringValue("missing");
}
return derivedField;
}
use of org.dmg.pmml.DerivedFieldDocument.DerivedField in project knime-core by knime.
the class PMMLGeneralRegressionTranslator method updateVectorLengthsBasedOnDerivedFields.
/**
* @param derivedFieldList
*/
private void updateVectorLengthsBasedOnDerivedFields(final List<DerivedField> derivedFieldList) {
final Map<String, Integer> lengths = new LinkedHashMap<>();
for (final DerivedField df : derivedFieldList) {
final String name = df.getName();
Optional<NameAndIndex> vni = VectorHandling.parse(name);
if (vni.isPresent()) {
final String key = vni.get().getName();
try {
String function = df.getApply().getFunction();
if (!"substring".equals(function)) {
continue;
}
final List<FieldRef> fieldRefList = df.getApply().getFieldRefList();
if (fieldRefList.isEmpty() || !key.equals(fieldRefList.get(0).getField())) {
LOGGER.debug("Field name is not related to the derived field name: " + fieldRefList + " <-> " + key);
continue;
}
if (2 != df.getApply().getConstantList().size()) {
LOGGER.debug("substring requires two parameters: " + df);
continue;
}
if (!DATATYPE.INTEGER.equals(df.getDataType())) {
LOGGER.debug("Array value should be integer: " + df);
continue;
}
if (!OPTYPE.CONTINUOUS.equals(df.getOptype())) {
LOGGER.debug("The optype should be continuous: " + df);
continue;
}
int index = vni.get().getIndex();
int old = Math.max(0, lengths.getOrDefault(key, Integer.valueOf(0)).intValue());
if (old <= index) {
lengths.put(key, index + 1);
}
} catch (RuntimeException e) {
// Ignore
LOGGER.debug(df.toString(), e);
}
}
}
LOGGER.debug(lengths);
m_content.updateVectorLengths(lengths);
}
use of org.dmg.pmml.DerivedFieldDocument.DerivedField in project knime-core by knime.
the class PMMLNeuralNetworkTranslator method addInputLayer.
/**
* Writes the PMML input layer of the MLP.
*
* @param nnModel
* the Neural Network model.
* @param mlp
* the underlying {@link MultiLayerPerceptron}.
*/
protected void addInputLayer(final NeuralNetwork nnModel, final MultiLayerPerceptron mlp) {
Layer inputlayer = mlp.getLayer(0);
Perceptron[] inputperceptrons = inputlayer.getPerceptrons();
HashMap<String, Integer> inputmap = mlp.getInputMapping();
NeuralInputs neuralInputs = nnModel.addNewNeuralInputs();
neuralInputs.setNumberOfInputs(BigInteger.valueOf(inputperceptrons.length));
for (int i = 0; i < inputperceptrons.length; i++) {
NeuralInput neuralInput = neuralInputs.addNewNeuralInput();
neuralInput.setId(0 + "," + i);
// search corresponding input column
String colname = "";
for (Entry<String, Integer> e : inputmap.entrySet()) {
if (e.getValue().equals(i)) {
colname = e.getKey();
}
}
DerivedField df = neuralInput.addNewDerivedField();
df.setOptype(OPTYPE.CONTINUOUS);
df.setDataType(DATATYPE.DOUBLE);
FieldRef fieldRef = df.addNewFieldRef();
fieldRef.setField(m_nameMapper.getDerivedFieldName(colname));
}
}
Aggregations