AttributeError：无法在 <module '__main__'> 上获取属性 'DataFrameSelector'

Question

我正在尝试加载腌制模型，但收到此错误：

model = joblib.load('pred.pkl')
AttributeError: module '__main__' has no attribute 'DataFrameSelector'

这是转储模型的方法：

from sklearn.externals import joblib
joblib.dump(best_rf, 'leadpred.pkl')

有什么想法吗？

Answer 1

执行模型的文件需要导入sklearn。