2.创建文件目录,保存模型
import os
from sklearn.externals import joblib
# 创建文件目录
dirs = 'testModel'
if not os.path.exists(dirs):
os.makedirs(dirs)
# 保存模型
joblib.dump(LR, dirs+'/LR.pkl')
3.读取模型
# 读取模型
LR = joblib.load(dirs+'/LR.pkl')
test = np.array([[3,4,5],[8,7,6]])
print('预测结果:\n', LR.predict(test))
python保存和调用模型
最新推荐文章于 2024-05-10 19:55:47 发布