GridSearchCV只保存一个估算器

Question

如何只保存一个GridSearchCV的估算器？当我使用“joblib.dump”函数时，它为“best_estimator_”生成了数百个文件。

model=cv.best_estimator_ 
joblib.dump(model, 'model.pkl') 

Answer 1

您正在做正确的事情。

的理由让你得到很多.pkl文件是

注：joblib.dump返回文件名列表。包含在clf对象中的每个单独的numpy数组被序列化为文件系统上的单独文件。使用joblib.load重新加载模型时，所有文件都需要在同一个文件夹中。

可在http://scikit-learn.org/stable/modules/model_persistence.html#persistence-example