小编本职工作Java,Python小白,现在需要调用Python机器训练模型,举一个模型为例子,目前已经生成pkl文件,并且可以在python调用。 现在需要由java发起调用,经过研究,可以生产pmml文件提供java调用,但是浏览无数博客均不顺利,希望各位走过路过帮帮小编瞅瞅,指点指点!!!!
ps:目前浏览其他博客生成pmml的python代码基本上都报错 !! 😓😓😓😓
from sklearn import linear_model as lm
import numpy as np
x = np.array([[12],[5],[7],[8]])
y = np.array([24,10,14,16])
model = lm.LinearRegression()
model.fit(x,y)
result = model.predict([[30]])
print(result)
import pickle as pk
将训练好的模型对象保存到磁盘文件
with open('/Volumes/Cute/linear.pkl','wb') as f:
pk.dump(model,f)
import pickle as pk
从磁盘文件加载模型对象
with open('/Volumes/Cute/linear.pkl','rb') as f:
model = pk.load(f)
根据输入预测输出
pred_y = model.predict([[10]])
print(pred_y)