protected String formRuleName(
PMML pmmlDocument, String modelName, Characteristic c, Attribute scoreAttribute) {
StringBuilder sb = new StringBuilder();
sb.append(modelName).append("_").append(c.getName()).append("_");
String dataType =
ScorecardPMMLUtils.getDataType(
pmmlDocument, ScorecardPMMLUtils.extractFieldNameFromCharacteristic(c));
if (XLSKeywords.DATATYPE_NUMBER.equalsIgnoreCase(dataType)) {
if (scoreAttribute.getSimplePredicate() != null) {
sb.append(scoreAttribute.getSimplePredicate().getOperator())
.append("_")
.append(scoreAttribute.getSimplePredicate().getValue());
} else if (scoreAttribute.getCompoundPredicate() != null) {
sb.append("between");
for (Object obj :
scoreAttribute
.getCompoundPredicate()
.getSimplePredicatesAndCompoundPredicatesAndSimpleSetPredicates()) {
if (obj instanceof SimplePredicate) {
sb.append("_").append(((SimplePredicate) obj).getValue());
}
}
}
} else if (XLSKeywords.DATATYPE_TEXT.equalsIgnoreCase(dataType)
|| XLSKeywords.DATATYPE_BOOLEAN.equalsIgnoreCase(dataType)) {
if (scoreAttribute.getSimplePredicate() != null) {
sb.append(scoreAttribute.getSimplePredicate().getOperator())
.append("_")
.append(scoreAttribute.getSimplePredicate().getValue());
} else if (scoreAttribute.getSimpleSetPredicate() != null) {
SimpleSetPredicate predicate = scoreAttribute.getSimpleSetPredicate();
Array array = predicate.getArray();
String text = array.getContent().replace(" ", "_");
sb.append(predicate.getBooleanOperator()).append("_").append(text);
}
}
return sb.toString();
}
protected void createFieldRestriction(
PMML pmmlDocument, Characteristic c, Attribute scoreAttribute, StringBuilder stringBuilder) {
stringBuilder.append("(");
// String dataType = ScorecardPMMLUtils.getExtensionValue(c.getExtensions(),
// PMMLExtensionNames.CHARACTERTISTIC_DATATYPE);
String dataType =
ScorecardPMMLUtils.getDataType(
pmmlDocument, ScorecardPMMLUtils.extractFieldNameFromCharacteristic(c));
if (XLSKeywords.DATATYPE_TEXT.equalsIgnoreCase(dataType)) {
if (scoreAttribute.getSimplePredicate() != null) {
SimplePredicate predicate = scoreAttribute.getSimplePredicate();
String operator = predicate.getOperator();
if (PMMLOperators.EQUAL.equalsIgnoreCase(operator)) {
stringBuilder.append(predicate.getField());
stringBuilder.append(" == ");
stringBuilder.append("\"").append(predicate.getValue()).append("\"");
} else if (PMMLOperators.NOT_EQUAL.equalsIgnoreCase(operator)) {
stringBuilder.append(predicate.getField());
stringBuilder.append(" != ");
stringBuilder.append("\"").append(predicate.getValue()).append("\"");
}
} else if (scoreAttribute.getSimpleSetPredicate() != null) {
SimpleSetPredicate simpleSetPredicate = scoreAttribute.getSimpleSetPredicate();
String content = simpleSetPredicate.getArray().getContent();
content = content.replaceAll(" ", "\",\"");
stringBuilder
.append(simpleSetPredicate.getField())
.append(" in ( \"")
.append(content)
.append("\" )");
}
} else if (XLSKeywords.DATATYPE_BOOLEAN.equalsIgnoreCase(dataType)) {
if (scoreAttribute.getSimplePredicate() != null) {
SimplePredicate predicate = scoreAttribute.getSimplePredicate();
String operator = predicate.getOperator();
if (PMMLOperators.EQUAL.equalsIgnoreCase(operator)) {
stringBuilder.append(predicate.getField());
stringBuilder.append(" == ");
stringBuilder.append(predicate.getValue().toLowerCase());
} else if (PMMLOperators.NOT_EQUAL.equalsIgnoreCase(operator)) {
stringBuilder.append(predicate.getField());
stringBuilder.append(" != ");
stringBuilder.append(predicate.getValue().toLowerCase());
}
}
} else if (XLSKeywords.DATATYPE_NUMBER.equalsIgnoreCase(dataType)) {
if (scoreAttribute.getSimplePredicate() != null) {
SimplePredicate predicate = scoreAttribute.getSimplePredicate();
String operator = predicate.getOperator();
stringBuilder.append(predicate.getField());
if (PMMLOperators.LESS_THAN.equalsIgnoreCase(operator)) {
stringBuilder.append(" < ");
} else if (PMMLOperators.GREATER_THAN.equalsIgnoreCase(operator)) {
stringBuilder.append(" > ");
} else if (PMMLOperators.NOT_EQUAL.equalsIgnoreCase(operator)) {
stringBuilder.append(" <> ");
} else if (PMMLOperators.EQUAL.equalsIgnoreCase(operator)) {
stringBuilder.append(" == ");
} else if (PMMLOperators.GREATER_OR_EQUAL.equalsIgnoreCase(operator)) {
stringBuilder.append(" >= ");
} else if (PMMLOperators.LESS_OR_EQUAL.equalsIgnoreCase(operator)) {
stringBuilder.append(" <= ");
}
stringBuilder.append(predicate.getValue());
} else if (scoreAttribute.getCompoundPredicate() != null) {
CompoundPredicate predicate = scoreAttribute.getCompoundPredicate();
String field = null;
for (Object obj :
predicate.getSimplePredicatesAndCompoundPredicatesAndSimpleSetPredicates()) {
if (obj instanceof SimplePredicate) {
SimplePredicate simplePredicate = (SimplePredicate) obj;
String operator = simplePredicate.getOperator();
if (field == null) {
stringBuilder.append(simplePredicate.getField());
field = simplePredicate.getField();
} else {
stringBuilder.append(" && ");
}
if (PMMLOperators.LESS_THAN.equalsIgnoreCase(operator)) {
stringBuilder.append(" < ");
} else if (PMMLOperators.GREATER_THAN.equalsIgnoreCase(operator)) {
stringBuilder.append(" > ");
} else if (PMMLOperators.NOT_EQUAL.equalsIgnoreCase(operator)) {
stringBuilder.append(" <> ");
} else if (PMMLOperators.EQUAL.equalsIgnoreCase(operator)) {
stringBuilder.append(" == ");
} else if (PMMLOperators.GREATER_OR_EQUAL.equalsIgnoreCase(operator)) {
stringBuilder.append(" >= ");
} else if (PMMLOperators.LESS_OR_EQUAL.equalsIgnoreCase(operator)) {
stringBuilder.append(" <= ");
}
stringBuilder.append(simplePredicate.getValue());
}
}
}
}
stringBuilder.append(")");
}