sklearn.externals.joblib不加载pkl文件

时间:2015-05-26 16:49:30

标签: python scikit-learn pickle joblib

我从一位离开我公司的同事那里继承了一些代码,我注意到他写的电子邮件打开预测器在几周内没有运行。所以,我试着运行它。它走到了这一行:

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很新,所以我不知道这是否会造成问题。感谢你能给出的任何指示

1 个答案:

答案 0 :(得分:0)

假设它适用于您的同事,并且您的问题是由于sklearn升级。 Splitter init的额外参数是在this commit中引入的。您可以尝试在早期版本的sklearn中加载pkl文件并从那里获取它。