调用模型

提示: 本文是模型部署方案的一部分

依赖

  • java8
  • pmml-evaluator 1.4.2
  • pmml-evaluator-extension 1.4.2

步骤

  1. 使用JPMML执行引擎反序列化pmml文件
  2. 模型校验（此处具体校验的问题，笔者以后另开一篇）

我们开始吧

maven依赖：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
<properties>
    <pmml-evaluator.version>1.4.2</pmml-evaluator.version>
</properties>

<!--运行模型-->
<dependency>
    <groupId>org.jpmml</groupId>
    <artifactId>pmml-evaluator</artifactId>
    <version>${pmml-evaluator.version}</version>
</dependency>
<dependency>
    <groupId>org.jpmml</groupId>
    <artifactId>pmml-evaluator-extension</artifactId>
    <version>${pmml-evaluator.version}</version>
</dependency>

一网打尽

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
package org.houqian.jpmml;

import org.jpmml.evaluator.*;
import org.xml.sax.SAXException;

import javax.xml.bind.JAXBException;
import java.io.File;
import java.io.IOException;
import java.util.List;

/**
 * @author : houqian
 * @version : 1.0
 * @since : 2018-08-30
 */
public class Execute {
  public static void main(String[] args) throws JAXBException, SAXException, IOException {
    // 从pmml文件实例化模型执行引擎
    Evaluator evaluator = new LoadingModelEvaluatorBuilder()
            .setLocatable(false)
            .setVisitors(new DefaultVisitorBattery())
            //.setOutputFilter(OutputFilters.KEEP_FINAL_RESULTS)
            .load(new File("src/main/resources/pipeline.pmml.xml"))
            .build();

    // 校验模型合法性
    evaluator.verify();

    // 输入变量
    List<? extends InputField> inputFields = evaluator.getInputFields();
    System.out.println("Input fields: " + inputFields);

    // 标签
    List<? extends TargetField> targetFields = evaluator.getTargetFields();
    System.out.println("Target field(s): " + targetFields);

    // 输出变量
    List<? extends OutputField> outputFields = evaluator.getOutputFields();
    System.out.println("Output fields: " + outputFields);

    // 释放引用，优化GC
    evaluator = null;
  }
}

Comments

▲