跨平台机器学习实践小结(一)

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

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

一、问题来源:

如何在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
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值