首页 > 代码库 > 模型持久化
模型持久化
__author__ = ‘Oscar_Yang‘ #-*- coding= utf-8 -*- from sklearn import datasets iris = datasets.load_iris() from sklearn.naive_bayes import GaussianNB from sklearn.externals import joblib import pickle from matplotlib import pyplot as plt gnb = GaussianNB() y_pred = gnb.fit(iris.data, iris.target).predict(iris.data) print("Number of mislabeled points out of a total {} points : {}".format(iris.data.shape[0],(iris.target != y_pred).sum())) #python提供两种模型持久化的方法,第一种是joblib,一种是pickle:如下: #第一种:比如gnb是我们想要保存的模型 joblib.dump(gnb, ‘GaussianNB.model‘) #模型保存到本地 lr = joblib.load(‘GauaaianNB.model‘) #重新加载模型 """ 重新加载模型, 需要注意的是,这里执行joblib.dump()之后,有可能还会生成若干个以rf.model_XX.npy为命名格式的文件, 这有可能是用于保存模型中的系数等的二进制文件。 其具体生成的文件的个数还会随调用到的分类器的不同,以及分类器中迭代次数的参数的不同而变,有时候会生成几个,有时候会生成几百个。 """ #或者第二种方式 save = pickle.dumps(gnb) #模保存模型 model = pickle.loads(save)#重新导入模型
模型持久化
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。