Java调用机器学习训练包记录一下

前言

• 最近公司有个需求，需要对用户进行数据画像分析。
• 公司大数据组通过对线上用户数据进行分析后，通过机器学习用python做了一个训练模型pkl文件包。
• 要求我部门对用户数据进行分析计算。而我部门的项目都是使用Java进行开发的，所以就需要Java调用pkl训练模型包。
• 经过调研python的pkl训练模型包不能直接被Java调用，跨平台调用需要使用pmml格式文件，所以就让大数据部门依照已经生成的训练模型pkl文件，在次封装成一个pmml文件。

pmml格式

<?xml version="1.0" encoding="UTF-8" standalone="yes"?>
<PMML xmlns="http://www.dmg.org/PMML-4_3" xmlns:data="http://jpmml.org/jpmml-model/InlineTable" version="4.3">
   <Header>
      <Application name="JPMML-SkLearn" version="1.6.27"/>
      <Timestamp>2021-08-30T06:48:45Z</Timestamp>
   </Header>
   <DataDictionary>
      <DataField name="y" optype="categorical" dataType="integer">
         <Value value="0"/>
         <Value value="1"/>
      </DataField>
      <DataField name="x1" optype="continuous" dataType="double"/>
      <DataField name="x2" optype="continuous" dataType="double"/>
      <DataField name="x3" optype="continuous" dataType="double"/>
   </DataDictionary>
   <RegressionModel functionName="classification" algorithmName="sklearn.linear_model._logistic.LogisticRegression" normalizationMethod="logit">
      <MiningSchema>
         <MiningField name="y" usageType="target"/>