GridSearchCV只保存一个估计器

Question

如何只保存一个GridSearchCV的估计器?当我使用"joblib"时。函数，它为"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