如何仅保存GridSearchCV的一个估计器?当我使用”joblib.dump”函数时,它为”best_estimator_”生成了数百个文件。
model=cv.best_estimator_
joblib.dump(model, 'model.pkl')
回答:
您的做法是正确的。
您得到许多.pkl
文件的原因是
注意:joblib.dump返回一个文件名列表。clf对象中包含的每个单独的numpy数组都被序列化为文件系统上的一个单独文件。重新加载模型时,所有文件都需要在同一个文件夹中使用joblib.load。
可以在http://scikit-learn.org/stable/modules/model_persistence.html#persistence-example找到相关信息