机器学习模型的保存和读取

最新推荐文章于 2024-02-16 22:01:08 发布

光明有我16620122910

最新推荐文章于 2024-02-16 22:01:08 发布

阅读量587

点赞数

文章标签： python

本文链接：https://blog.csdn.net/u011619323/article/details/109300791

版权

import os
from sklearn.ensemble import RandomForestClassifier
from sklearn import datasets
from sklearn.metrics import accuracy_score
import pickle

from sklearn.model_selection import train_test_split

sheet = datasets.load_wine()
x,y=sheet.data,sheet.target
clf = RandomForestClassifier(n_estimators=14,random_state=1,max_depth=5)
xtrain,xtest,ytrain,ytest=train_test_split(x,y,test_size=0.2,random_state=1)
clf.fit(xtrain,ytrain)
# 保存模型
file_path = os.path.dirname(os.path.realpath(__file__))
with open ('clf.pickle', 'wb') as f:
	pickle.dump(clf,f)
# 读取模型
with open ('clf.pickle', 'rb') as f:
	clf = pickle.load(f)
	y_pred=clf.predict(xtest)
	print(accuracy_score(ytest,y_pred))

```python
在这里插入代码片