我从一位离开公司的同事那里继承了一些代码,我注意到他写的一个电子邮件打开预测器已经几个星期没有运行了。所以,我试着运行它。它到了这一行:
clf_predict = joblib.load(os.path.join(data_dir,'trained_random_forest.pkl'))
结果是:
File "/usr/local/lib/python2.7/distpackages/sklearn/externals/joblib/numpy_pickle.py", line 425, in load
obj = unpickler.load()
File "/usr/lib/python2.7/pickle.py", line 858, in load dispatch[key](self)
File "/usr/lib/python2.7/pickle.py", line 1133, in load_reduce
value = func(*args)
File "sklearn/tree/_tree.pyx", line 923, in sklearn.tree._tree.Splitter.__cinit__ (sklearn/tree/_tree.c:8329)
TypeError: __cinit__() takes exactly 5 positional arguments (4 given)
从我在文档中看到的来看,加载的语法是正确的。因此,我最初的理论是scikit-learn可能已经升级,无法读取当前的pickle文件(此时该文件已经存在几个月了)。然而,我不确定它是否升级了,而且我对使用sklearn非常陌生,所以我不知道这是否会造成问题。
假设它对你的同事有效,你的问题是由于sklearn升级。Splitter init的额外参数是在这次提交中引入的。您可以尝试在早期版本的sklearn中加载pkl文件并从那里获取。