sklearn 的模型保存与加载使用

时间:2025-02-23 15:10:30
'''
训练好了模型之后保存到本地,并重新加载模型用于预测的两种方法
'''
# 方法1:pickle
from sklearn import svm
from sklearn import datasets 
import pickle 
clf = ()
iris = datasets.load_iris() 
X,y = ,  
(X,y)
# 模型的保存
with open('','wb') as f: 
    (clf,f) #将训练好的模型clf存储在变量f中,且保存到本地

# 模型的重新加载和使用
with open('','rb') as f:  
    clf_load = (f)  #将模型存储在变量clf_load中  
    print(clf_load.predict(X[0:1000])) #调用模型并预测结果
# 方法2:joblib方法
from sklearn import svm 
from sklearn import datasets
import joblib
# 函数是用在0.21及以前的版本中,在最新的版本中,该函数应被弃用改为直接导入joblib
# from  import joblib
clf = ()
iris = datasets.load_iris() 
X,y = ,  
(X,y) 
# 保存训练好的clf模型 
(clf,'',compress=3) 
# 重新加载训练好的clf模型 
clf3 = ('')  
print((X[0:1000]))  # 打印预测值
# 相对比而言,joblib更高效,但32位python和64位python不能无缝连接使用,可以通过如下代码查看本机的情况
import platform 
print(())