将svm训练好的模型保存为model.pkl文件python代码实现
时间: 2024-05-09 14:15:39 浏览: 212
基于SVM和LSTM实现的购物平台商品评论情感对比分析源码+全部数据(python毕设).zip
以下是一个简单的示例代码,用于将训练好的SVM模型保存为名为model.pkl的文件:
``` python
import pickle
from sklearn import svm
from sklearn import datasets
# 加载数据集
iris = datasets.load_iris()
X, y = iris.data, iris.target
# 训练模型
clf = svm.SVC()
clf.fit(X, y)
# 保存模型
with open('model.pkl', 'wb') as f:
pickle.dump(clf, f)
```
上述代码中,我们使用pickle模块将训练好的SVM模型保存到名为'model.pkl'的文件中。您可以根据需要更改文件名和模型变量名。
阅读全文