use of org.dmg.pmml.TransformationDictionaryDocument.TransformationDictionary in project knime-core by knime.
the class PMMLStringConversionTranslator method exportToTransDict.
/**
* {@inheritDoc}
*/
@Override
public TransformationDictionary exportToTransDict() {
TransformationDictionary dictionary = TransformationDictionary.Factory.newInstance();
dictionary.setDerivedFieldArray(createDerivedFields());
return dictionary;
}
use of org.dmg.pmml.TransformationDictionaryDocument.TransformationDictionary in project knime-core by knime.
the class PMMLBinningTranslator method exportToTransDict.
/**
* {@inheritDoc}
*/
@Override
public TransformationDictionary exportToTransDict() {
TransformationDictionary dictionary = TransformationDictionary.Factory.newInstance();
dictionary.setDerivedFieldArray(createDerivedFields());
return dictionary;
}
use of org.dmg.pmml.TransformationDictionaryDocument.TransformationDictionary in project knime-core by knime.
the class PMMLMany2OneTranslator method exportToTransDict.
/**
* {@inheritDoc}
*/
@Override
public TransformationDictionary exportToTransDict() {
final TransformationDictionary dictionary = TransformationDictionary.Factory.newInstance();
dictionary.setDerivedFieldArray(new DerivedField[] { createDerivedField() });
return dictionary;
}
use of org.dmg.pmml.TransformationDictionaryDocument.TransformationDictionary in project knime-core by knime.
the class PMMLMapValuesTranslator method exportToTransDict.
/**
* {@inheritDoc}
*/
@Override
public TransformationDictionary exportToTransDict() {
TransformationDictionary dictionary = TransformationDictionary.Factory.newInstance();
dictionary.setDerivedFieldArray(createDerivedFields());
return dictionary;
}
use of org.dmg.pmml.TransformationDictionaryDocument.TransformationDictionary in project knime-core by knime.
the class DBAutoBinner method translate.
/**
* This method translates a {@link PMMLPreprocDiscretize} object into {@link PMMLPortObject}.
*
* @param pmmlDiscretize {@link PMMLPreprocDiscretize} object
* @param dataTableSpec {@link DataTableSpec} if incoming {@link BufferedDataTable}
* @return a {@link PMMLPortObject} containing required parameters for binning operation
*/
public static PMMLPortObject translate(final PMMLPreprocDiscretize pmmlDiscretize, final DataTableSpec dataTableSpec) {
final Map<String, Bin[]> columnToBins = new HashMap<>();
final Map<String, String> columnToAppend = new HashMap<>();
List<String> replacedColumnNames = pmmlDiscretize.getConfiguration().getNames();
for (String replacedColumnName : replacedColumnNames) {
PMMLDiscretize discretize = pmmlDiscretize.getConfiguration().getDiscretize(replacedColumnName);
List<PMMLDiscretizeBin> bins = discretize.getBins();
String originalColumnName = discretize.getField();
if (replacedColumnName.equals(originalColumnName)) {
// wenn replaced, dann nicht anhängen
columnToAppend.put(originalColumnName, null);
} else {
// nicht replaced -> anhängen
columnToAppend.put(originalColumnName, replacedColumnName);
}
NumericBin[] numericBin = new NumericBin[bins.size()];
int counter = 0;
for (PMMLDiscretizeBin bin : bins) {
String binName = bin.getBinValue();
List<PMMLInterval> intervals = bin.getIntervals();
boolean leftOpen = false;
boolean rightOpen = false;
double leftMargin = 0;
double rightMargin = 0;
// always returns only one interval
for (PMMLInterval interval : intervals) {
Closure closure = interval.getClosure();
switch(closure) {
case openClosed:
leftOpen = true;
rightOpen = false;
break;
case openOpen:
leftOpen = true;
rightOpen = true;
break;
case closedOpen:
leftOpen = false;
rightOpen = true;
case closedClosed:
leftOpen = false;
rightOpen = false;
break;
default:
leftOpen = true;
rightOpen = false;
break;
}
leftMargin = interval.getLeftMargin();
rightMargin = interval.getRightMargin();
}
numericBin[counter] = new NumericBin(binName, leftOpen, leftMargin, rightOpen, rightMargin);
counter++;
}
columnToBins.put(originalColumnName, numericBin);
}
// ColumnRearranger createColReg = createColReg(dataTableSpec, columnToBins, columnToAppended);
DataTableSpec newDataTableSpec = createNewDataTableSpec(dataTableSpec, columnToAppend);
PMMLPortObjectSpecCreator pmmlSpecCreator = new PMMLPortObjectSpecCreator(newDataTableSpec);
PMMLPortObject pmmlPortObject = new PMMLPortObject(pmmlSpecCreator.createSpec(), null, newDataTableSpec);
PMMLBinningTranslator trans = new PMMLBinningTranslator(columnToBins, columnToAppend, new DerivedFieldMapper(pmmlPortObject));
TransformationDictionary exportToTransDict = trans.exportToTransDict();
pmmlPortObject.addGlobalTransformations(exportToTransDict);
return pmmlPortObject;
}
Aggregations