python 模型的保存

2023-01-31 01:01:37 python 模型 保存

这里写图片描述

来源于知乎

模型上线一般通过java处理
此时最好用pmml,GitHub上有sklearntopmml的模块可以免费使用,强烈推荐。
这和R语言有点类似

完整的一个例子

# conding = utf-8
from sklearn import svm
from sklearn.externals import joblib
import os
X = [[0,1],[0,2],[0,3],[1,0],[2,0],[3,0]]
y= [0,0,0,1,1,1]
clf = svm.SVC(probability=True)
clf.fit(X,y)
# 测试样本
test = [[2.5,2.5],[2.5,0]]
result = clf.predict_proba(test)
print(result)

joblib.dump(clf,"train_model.m")
clf2 =joblib.load("train_model.m")
test2 =[[2.,2.],[1.,2.]]
result2 = clf2.predict_proba(test2)
print(result2)

相关文章