scikit-learn的基本用法——模型保存与加载

最新推荐文章于 2024-10-25 10:02:16 发布

转载最新推荐文章于 2024-10-25 10:02:16 发布 · 1.1k 阅读

Python 同时被 2 个专栏收录

81 篇文章

订阅专栏

Machine Learning & Data Mining

33 篇文章

订阅专栏

本文通过两个示例介绍了如何使用scikit-learn保存和加载机器学习模型。示例一采用Python的pickle模块进行操作；示例二则利用了scikit-learn自带的joblib工具。两个示例均使用了鸢尾花数据集进行模型训练。

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

本文主要介绍scikit-learn中的模型的保存与加载。

Demo 1

import pickle
from sklearn.svm import SVC
from sklearn import datasets

# 定义分类器
svm = SVC()

# 加载iris数据集
iris = datasets.load_iris()
# 读取特征
X = iris.data
# 读取分类标签
y = iris.target

# 训练模型
svm.fit(X, y)

# 保存成python支持的文件格式pickle, 在当前目录下可以看到svm.pickle
with open('svm.pickle', 'wb') as fw:
    pickle.dump(svm, fw)

# 加载svm.pickle
with open('svm.pickle', 'rb') as fr:
    new_svm = pickle.load(fr)
    print new_svm.predict(X[0:1])
 1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25

结果

[0]
 1

Demo 2

from sklearn.externals import joblib
from sklearn.svm import SVC
from sklearn import datasets

# 定义分类器
svm = SVC()

# 加载iris数据集
iris = datasets.load_iris()
# 读取特征
X = iris.data
# 读取分类标签
y = iris.target

# 训练模型
svm.fit(X, y)

# 保存成sklearn自带的文件格式
joblib.dump(svm, 'svm.pkl')

# 加载svm.pkl
new_svm = joblib.load('svm.pkl')
print new_svm.predict(X[0:1])
 1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23