Examples with KiePMMLMiningField org.kie.pmml.commons.model.KiePMMLMiningField used on opensource projects

Example 16 with KiePMMLMiningField

use of org.kie.pmml.commons.model.KiePMMLMiningField in project drools by kiegroup.

the class PreProcessTest method convertInputDataConvertibles.

@Test
public void convertInputDataConvertibles() {
    List<KiePMMLMiningField> miningFields = IntStream.range(0, 3).mapToObj(i -> {
        DATA_TYPE dataType = DATA_TYPE.values()[i];
        return KiePMMLMiningField.builder("FIELD-" + i, null).withDataType(dataType).build();
    }).collect(Collectors.toList());
    PMMLRequestData pmmlRequestData = new PMMLRequestData("123", "modelName");
    pmmlRequestData.addRequestParam("FIELD-0", 123);
    pmmlRequestData.addRequestParam("FIELD-1", "123");
    pmmlRequestData.addRequestParam("FIELD-2", "1.23");
    Map<String, ParameterInfo> mappedRequestParams = pmmlRequestData.getMappedRequestParams();
    assertEquals(123, mappedRequestParams.get("FIELD-0").getValue());
    assertEquals("123", mappedRequestParams.get("FIELD-1").getValue());
    assertEquals("1.23", mappedRequestParams.get("FIELD-2").getValue());
    PreProcess.convertInputData(miningFields, pmmlRequestData);
    assertEquals("123", mappedRequestParams.get("FIELD-0").getValue());
    assertEquals(123, mappedRequestParams.get("FIELD-1").getValue());
    assertEquals(1.23f, mappedRequestParams.get("FIELD-2").getValue());
}

Example 17 with KiePMMLMiningField

use of org.kie.pmml.commons.model.KiePMMLMiningField in project drools by kiegroup.

the class PreProcessTest method manageMissingValuesReturnInvalid.

@Test(expected = KiePMMLException.class)
public void manageMissingValuesReturnInvalid() {
    KiePMMLMiningField miningField = KiePMMLMiningField.builder("FIELD", null).withDataType(DATA_TYPE.STRING).withMissingValueTreatmentMethod(MISSING_VALUE_TREATMENT_METHOD.RETURN_INVALID).build();
    PreProcess.manageMissingValues(miningField, new PMMLRequestData());
}

Example 18 with KiePMMLMiningField

use of org.kie.pmml.commons.model.KiePMMLMiningField in project drools by kiegroup.

the class PreProcessTest method verifyFixInvalidValuesInvalidAsValueWithReplacement.

@Test
public void verifyFixInvalidValuesInvalidAsValueWithReplacement() {
    KiePMMLMiningField miningField0 = KiePMMLMiningField.builder("FIELD-0", null).withDataType(DATA_TYPE.STRING).withInvalidValueTreatmentMethod(INVALID_VALUE_TREATMENT_METHOD.AS_VALUE).withInvalidValueReplacement("123").withAllowedValues(Arrays.asList("123", "124", "125")).build();
    KiePMMLMiningField miningField1 = KiePMMLMiningField.builder("FIELD-1", null).withDataType(DATA_TYPE.DOUBLE).withInvalidValueTreatmentMethod(INVALID_VALUE_TREATMENT_METHOD.AS_VALUE).withInvalidValueReplacement("1.23").withAllowedValues(Arrays.asList("1.23", "12.4", "1.25")).build();
    List<KiePMMLInterval> intervals = Arrays.asList(new KiePMMLInterval(0.0, 12.4, CLOSURE.CLOSED_CLOSED), new KiePMMLInterval(12.6, 14.5, CLOSURE.OPEN_CLOSED));
    KiePMMLMiningField miningField2 = KiePMMLMiningField.builder("FIELD-2", null).withDataType(DATA_TYPE.DOUBLE).withInvalidValueTreatmentMethod(INVALID_VALUE_TREATMENT_METHOD.AS_VALUE).withInvalidValueReplacement("12.3").withIntervals(intervals).build();
    List<KiePMMLMiningField> miningFields = Arrays.asList(miningField0, miningField1, miningField2);
    PMMLRequestData pmmlRequestData = new PMMLRequestData("123", "modelName");
    pmmlRequestData.addRequestParam("FIELD-0", "122");
    pmmlRequestData.addRequestParam("FIELD-1", 12.5);
    pmmlRequestData.addRequestParam("FIELD-2", 14.6);
    PreProcess.verifyFixInvalidValues(miningFields, pmmlRequestData);
    Map<String, ParameterInfo> mappedRequestParams = pmmlRequestData.getMappedRequestParams();
    assertEquals("123", mappedRequestParams.get("FIELD-0").getValue());
    assertEquals(1.23, mappedRequestParams.get("FIELD-1").getValue());
    assertEquals(12.3, mappedRequestParams.get("FIELD-2").getValue());
}

Example 19 with KiePMMLMiningField

use of org.kie.pmml.commons.model.KiePMMLMiningField in project drools by kiegroup.

the class PreProcessTest method verifyAddMissingValuesMissingReturnInvalid.

@Test(expected = KiePMMLException.class)
public void verifyAddMissingValuesMissingReturnInvalid() {
    List<KiePMMLMiningField> miningFields = IntStream.range(0, 3).mapToObj(i -> {
        DATA_TYPE dataType = DATA_TYPE.values()[i];
        return KiePMMLMiningField.builder("FIELD-" + i, null).withDataType(dataType).withMissingValueTreatmentMethod(MISSING_VALUE_TREATMENT_METHOD.RETURN_INVALID).build();
    }).collect(Collectors.toList());
    PMMLRequestData pmmlRequestData = new PMMLRequestData("123", "modelName");
    PreProcess.verifyAddMissingValues(miningFields, pmmlRequestData);
}

Example 20 with KiePMMLMiningField

use of org.kie.pmml.commons.model.KiePMMLMiningField in project drools by kiegroup.

the class PreProcessTest method manageInvalidValuesNotReturnInvalid.

@Test
public void manageInvalidValuesNotReturnInvalid() {
    final ParameterInfo parameterInfo = new ParameterInfo();
    // AS_MISSING
    KiePMMLMiningField miningField = KiePMMLMiningField.builder("FIELD", null).withInvalidValueTreatmentMethod(INVALID_VALUE_TREATMENT_METHOD.AS_MISSING).build();
    List<ParameterInfo> toRemove = new ArrayList<>();
    PreProcess.manageInvalidValues(miningField, parameterInfo, toRemove);
    assertEquals(1, toRemove.size());
    assertTrue(toRemove.contains(parameterInfo));
    // AS_IS
    miningField = KiePMMLMiningField.builder("FIELD", null).withInvalidValueTreatmentMethod(INVALID_VALUE_TREATMENT_METHOD.AS_IS).build();
    toRemove = new ArrayList<>();
    PreProcess.manageInvalidValues(miningField, parameterInfo, toRemove);
    assertTrue(toRemove.isEmpty());
    // AS_VALUE with replacement
    String invalidValueReplacement = "REPLACEMENT";
    miningField = KiePMMLMiningField.builder("FIELD", null).withDataType(DATA_TYPE.STRING).withInvalidValueTreatmentMethod(INVALID_VALUE_TREATMENT_METHOD.AS_VALUE).withInvalidValueReplacement(invalidValueReplacement).build();
    toRemove = new ArrayList<>();
    assertNull(parameterInfo.getValue());
    assertNull(parameterInfo.getType());
    PreProcess.manageInvalidValues(miningField, parameterInfo, toRemove);
    assertTrue(toRemove.isEmpty());
    assertEquals(invalidValueReplacement, parameterInfo.getValue());
    assertEquals(String.class, parameterInfo.getType());
}