Java如何调⽤python平台的机器学习模型
1. 训练好模型后,保存模型
joblib.dump(xgb,"xgb_model.pkl")
2. 新建⼀个python⽂件,useModel.py。
import sys
if __name__ =='__main__':
model = joblib.load("xgb_model.pkl")# 加载模型
value=[]
python转java代码# 接收参数
for i in range(len(sys.argv)):
value.append(float(sys.argv[i]))
# 预测
predict = model.predict(value)
3. java可以通过调⽤python解释器,即使⽤jpython。但是此种⽅式必须保证python代码内不包含第三⽅库。如果要包含第三⽅库,需
要通过如下⽅式:
public static void main(String[] args)throws IOException, InterruptedException {
System.out.println("start python");
String[] arg =new String[]{"python","test\\useModel.py","123","4"};
//参数2:python⽂件路径;参数3,参数4:python代码的输⼊参数。
Process pr = Runtime().exec(arg);
BufferedReader in =new BufferedReader(new InputStream()));
String line;
while((line = in.readLine())!=null){
System.out.println(line);
}
in.close();
System.out.println("end");
pr.waitFor();
}
版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系QQ:729038198,我们将在24小时内删除。
发表评论