python保存和调用模型

最新推荐文章于 2025-04-08 10:39:55 发布

sphinxrascal168

最新推荐文章于 2025-04-08 10:39:55 发布

阅读量1.5k

点赞数

分类专栏：大幅度

本文链接：https://blog.youkuaiyun.com/sphinxrascal168/article/details/115273565

版权

大幅度专栏收录该内容

41 篇文章

订阅专栏

该博客介绍了如何使用Python的Scikit-Learn库进行模型的保存和加载。首先，通过`os`模块创建了一个名为`testModel`的文件夹，接着利用`joblib.dump()`将逻辑回归模型（LR）保存到该目录下。之后，使用`joblib.load()`成功读取模型，并对测试数据进行预测，展示了模型的实用性。

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

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))