from sklearn.ensemble import RandomForestClassifier
from sklearn import datasets
import joblib
(X,Y) = datasets.load_iris(return_X_y=True)
rfc = RandomForestClassifier(n_estimators=100,max_depth=100)
rfc.fit(X,Y)
print(rfc.predict(X[0:5,:]))
#保存模型
joblib.dump(rfc,'saved_model/rfc.pkl')
#读取模型
rfc_new = joblib.load('saved_model/rfc.pkl')
print(rfc_new.predict(X[147:150,:]))
保存机器学习训练的模型:使用sklearn中的joblib模块
最新推荐文章于 2024-06-25 11:50:09 发布