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。