python训练模型上线问题总结

最新推荐文章于 2024-08-09 07:19:03 发布

Dreamcatcher5

最新推荐文章于 2024-08-09 07:19:03 发布

阅读量4.5k

点赞数

分类专栏： python 文章标签： python java

本文链接：https://blog.csdn.net/qq_31955775/article/details/78329699

版权

本文总结了在Java中调用Python训练的模型所遇到的问题及解决方案。通过Runtime.getRuntime().exec(args)方法执行Python脚本，讨论了如何处理程序运行时信息输出导致的阻塞问题，提出了采用多线程打印信息的策略来避免阻塞。

摘要由CSDN通过智能技术生成

java调用python模型

PMML格式
使用java自带的Runtime.getRuntime().exec(args);方法直接调用python脚本

PMML格式

1、首先将python代码训练的模型保存为pmml格式，代码如下

model = xgb.XGBClassifier()
from sklearn2pmml import PMMLPipeline
pipeline = PMMLPipeline([("classifier", model)])
pipeline.fit(X_train,y_train)    
from sklearn2pmml import sklearn2pmml
sklearn2pmml(pipeline, "xgb.pmml", with_repr = True)

然后使用java读取pmml文件对数据进行预测，
后来选择使用java调用虚拟机的方式运行python脚本。

import java.io.IOException;
import java.io.InputStream;
import java.util.Map;
import javax.xml.bind.JAXBException;
import org.dmg.pmml.FieldName;
import org.dmg.pmml.PMML;
import org.jpmml.evaluator.ModelEvaluator;
import org.jpmml.evaluator.ModelEvaluatorFactory;
import org.jpmml.model.PMMLUtil;
import org.xml.sax.SAXException;

/**
 * 读取pmml 获取模型
 * 
 * @author liaotuo
 *
 */
public class ModelInvoker {
   
    private ModelEvaluator modelEvaluator;

    // 通过文件读取模型
    public ModelInvoker(String pmmlFileName) {
        PMML pmml = null;
        InputStream is = null;
        try {
            if (pmmlFileName != null) {
                is = ModelInvoker.class.getClassLoader().getResourceAsStream(pmmlFileName);
                pmml = PMMLUtil.unmarshal(is);
            }
            try {
                is.close();
            } <