一、问题来源:
如何在node web服务下调用sklearn的模型结果来进行实时模型预测?
二、问题分析:
1、sklearn的模型结果有几种存储方式:
(1)pickle.dumps ,结果通过变量保存在内存中
附上pickle文档:https://docs.python.org/2/library/pickle.html
>>> import pickle >>> s = pickle.dumps(clf) >>> clf2 = pickle.loads(s) >>> clf2.predict(X[0:1]) array([0])
(2)joblib.dump,持久化到二进制文件pkl中,可复用性更强
>>> from sklearn.externals import joblib >>> joblib.dump(clf, 'filename.pkl')
预测时,在另一个python进程中可以执行:
>>> clf2 = joblib.load('filename.pkl')
(3)pmml文件,PMML一种使用xml描述模型的语言标准
最有用的参考文档是:http://dmg.org/pmml/v4-1/GeneralStructure.html ,对PMML文件的结构及标签含义都有较清楚的说明。
先说明一下生成方式,以sklearn的gbdtregression为例:
from sklearn2pmml import sklearn2

本文探讨了在Node.js Web服务中调用sklearn模型预测的问题,分析了sklearn模型的存储方式,包括pickle和joblib,重点介绍了PMML作为跨平台标准的优势。通过JPMML库在Java中应用PMML,支持sklearn等框架。同时,文章指出了sklearn2pmml生成的PMML文件与Python图形库生成的dot文件在结构上的差异,但预测结果基本一致。
最低0.47元/天 解锁文章
1792

被折叠的 条评论
为什么被折叠?



