python - 保存sklearn交叉验证对象

Question

按照 sklearn 的教程进行操作，我尝试保存通过 sklearn 创建的对象，但没有成功。问题似乎出在交叉验证对象上，因为我可以保存实际的(最终)模型。

给定:

rf_model = RandomForestRegressor(n_estimators=1000, n_jobs=4, compute_importances = False)
cvgridsrch = GridSearchCV(estimator=rf_model, param_grid=parameters,n_jobs=4) 
cvgridsrch.fit(X,y)

这将会成功:

joblib.dump(cvgridsrch.best_estimator_, 'C:\\Users\\Desktop\\DMA\\cvgridsrch.pkl', compress=9)

这将会失败:

joblib.dump(cvgridsrch, 'C:\\Users\\Desktop\\DMA\\cvgridsrch.pkl', compress=9)

有错误:

PicklingError: Can't pickle <type 'instancemethod'>: it's not found as __builtin__.instancemethod

如何保存完整的对象？

Answer 1

如果您使用的是 Python 2，尝试:

import dill  

这样 lambda 函数就可以被 pickle....