在Java中處理PMML數(shù)據(jù),您需要使用一個(gè)支持PMML的庫(kù),例如JPMML或者OpenScore。下面是使用JPMML庫(kù)處理PMML數(shù)據(jù)的簡(jiǎn)單示例:
<groupId>org.jpmml</groupId>
<artifactId>pmml-evaluator</artifactId>
<version>1.5.13</version>
</dependency>
import org.dmg.pmml.FieldName;
import org.dmg.pmml.PMML;
import org.jpmml.evaluator.Evaluator;
import org.jpmml.evaluator.FieldValue;
import org.jpmml.evaluator.ModelEvaluatorFactory;
import org.jpmml.model.PMMLUtil;
import java.io.FileInputStream;
import java.io.InputStream;
import java.util.HashMap;
import java.util.Map;
public static Evaluator createEvaluator(String pmmlFilePath) throws Exception {
InputStream inputStream = new FileInputStream(pmmlFilePath);
PMML pmml = PMMLUtil.unmarshal(inputStream);
return ModelEvaluatorFactory.newInstance().newModelEvaluator(pmml);
}
public static Map<FieldName, ?> evaluateModel(Evaluator evaluator, Map<String, Object> inputData) {
Map<FieldName, FieldValue> arguments = new HashMap<>();
for (Map.Entry<String, Object> entry : inputData.entrySet()) {
FieldName fieldName = evaluator.getInputField(entry.getKey()).getName();
FieldValue fieldValue = evaluator.prepare(fieldName, entry.getValue());
arguments.put(fieldName, fieldValue);
}
Map<FieldName, ?> results = evaluator.evaluate(arguments);
return results;
}
public static void main(String[] args) {
try {
String pmmlFilePath = "path/to/your/pmml/file.pmml";
Evaluator evaluator = createEvaluator(pmmlFilePath);
Map<String, Object> inputData = new HashMap<>();
inputData.put("input_field_name_1", value1);
inputData.put("input_field_name_2", value2);
// ... 添加其他輸入字段和值
Map<FieldName, ?> results = evaluateModel(evaluator, inputData);
System.out.println("Predicted value: " + results.get(evaluator.getTargetField().getName()));
} catch (Exception e) {
e.printStackTrace();
}
}
將上述代碼中的path/to/your/pmml/file.pmml
替換為您的PMML文件路徑,并根據(jù)您的模型添加輸入字段和值。運(yùn)行程序后,您將看到模型的預(yù)測(cè)結(jié)果。