保存训练好的机器学习模型
训练好一个机器学习模型后,下次如果还想用这个模型,就需要把这个模型保存下来,下次直接导入就即可,不然每次都跑一遍,sklearn官网提供了两种保存model的方法:
1.使用python自带的pickle
from sklearn.ensemble import RandomForestClassifier
from sklearn import datasets
import pickle
(X,y) = datasets.load_iris(return_X_y=True)
rfc = RandomForestClassifier(n_estimators=100,max_depth=100)
rfc.fit(X,y)
print(