Examples with MiningField org.kie.dmg.pmml.pmml_4_2.descr.MiningField used on opensource projects

Example 6 with MiningField

use of org.kie.dmg.pmml.pmml_4_2.descr.MiningField in project drools by kiegroup.

the class PMML4UnitImpl method getMiningFields.

/**
 * Retrieves a Map with entries that consist of
 * key -> a model identifier
 * value -> the List of raw MiningField objects belonging to the model referenced by the key
 * @return The Map of model identifiers and their corresponding list of raw MiningField objects
 */
public Map<String, List<MiningField>> getMiningFields() {
    Map<String, List<MiningField>> miningFieldsMap = new HashMap<>();
    for (PMML4Model model : getModels()) {
        List<MiningField> miningFields = model.getRawMiningFields();
        miningFieldsMap.put(model.getModelId(), miningFields);
        model.getChildModels();
    }
    return miningFieldsMap;
}