在Java中集成PMML(Predictive Model Markup Language)工具,可以使用開源庫如JPMML和OpenNLP
首先,您需要將JPMML的相關依賴項添加到項目中。如果您使用Maven或Gradle構建項目,請將以下依賴項添加到pom.xml
或build.gradle
文件中。
Maven:
<dependency>
<groupId>org.jpmml</groupId>
<artifactId>pmml-evaluator</artifactId>
<version>1.5.13</version>
</dependency>
<dependency>
<groupId>org.jpmml</groupId>
<artifactId>pmml-model</artifactId>
<version>1.5.13</version>
</dependency>
</dependencies>
Gradle:
dependencies {
implementation 'org.jpmml:pmml-evaluator:1.5.13'
implementation 'org.jpmml:pmml-model:1.5.13'
}
從文件系統或URL加載PMML模型。例如:
import org.dmg.pmml.PMML;
import org.jpmml.model.PMMLUtil;
File pmmlFile = new File("path/to/your/pmml/file.pmml");
PMML pmml = PMMLUtil.unmarshal(pmmlFile);
使用JPMML庫創建一個Evaluator實例,該實例將用于評估PMML模型。
import org.jpmml.evaluator.ModelEvaluator;
import org.jpmml.evaluator.ModelEvaluatorFactory;
ModelEvaluator<?> evaluator = ModelEvaluatorFactory.newInstance().newModelEvaluator(pmml);
為模型評估準備輸入數據。例如:
import org.jpmml.evaluator.Evaluator;
import org.jpmml.evaluator.InputField;
Map<String, ?> arguments = new LinkedHashMap<>();
List<InputField> inputFields = evaluator.getInputFields();
for (InputField inputField : inputFields) {
FieldName fieldName = inputField.getName();
Object value = // 獲取輸入值,例如從用戶輸入、數據庫或其他來源獲取
arguments.put(fieldName.getValue(), value);
}
使用Evaluator評估模型并獲取預測結果。
import org.jpmml.evaluator.EvaluationContext;
import org.jpmml.evaluator.EvaluatorUtil;
import org.jpmml.evaluator.OutputField;
import org.jpmml.evaluator.TargetField;
EvaluationContext context = new EvaluationContext(evaluator);
context.setArguments(arguments);
Map<FieldName, ?> results = EvaluatorUtil.decodeAll(evaluator.evaluate(context));
處理模型評估的輸出結果。例如:
import org.jpmml.evaluator.FieldValue;
for (OutputField outputField : evaluator.getOutputFields()) {
FieldName fieldName = outputField.getName();
Object value = results.get(fieldName);
if (value instanceof FieldValue) {
FieldValue fieldValue = (FieldValue) value;
System.out.println(fieldName + ": " + fieldValue.getValue());
} else {
System.out.println(fieldName + ": " + value);
}
}
這樣,您就可以在Java項目中集成PMML工具,評估和使用PMML模型進行預測了。