sklearn训练好的模型导出代码
时间: 2023-08-23 11:05:39 浏览: 86
基于sklearn的一些机器学习的代码
你可以使用Python中的pickle模块将sklearn训练好的模型导出为二进制文件,以便以后可以重新加载模型并进行预测。
下面是一个例子:
```python
import pickle
from sklearn.ensemble import RandomForestClassifier
# 训练模型
X = [[0, 0], [1, 1]]
y = [0, 1]
clf = RandomForestClassifier()
clf.fit(X, y)
# 导出模型
with open('model.pkl', 'wb') as f:
pickle.dump(clf, f)
```
在上面的代码中,我们首先训练了一个随机森林分类器,然后使用pickle将模型导出为一个名为“model.pkl”的二进制文件。
要重新加载模型并进行预测,可以使用以下代码:
```python
import pickle
# 加载模型
with open('model.pkl', 'rb') as f:
clf = pickle.load(f)
# 进行预测
print(clf.predict([[2., 2.]]))
```
在上面的代码中,我们首先使用pickle加载保存的模型,然后对一个新样本进行预测并输出结果。
希望这个例子能够帮助您导出sklearn训练好的模型。
阅读全文