不能腌制Scikit学习NearestNeighbor分类器 - 不能pickle instancemethod对象

Question

我正在尝试挑选NearestNeighbor模型，但它说不能腌制instancemethod对象。

代码：

import cPickle as pickle
from sklearn.neighbors import NearestNeighbors

nbrs = NearestNeighbors(n_neighbors=50, algorithm='ball_tree', metric=self.distanceCIE2000_classifier)
nbrs.fit(allValues)
with open('/home/ubuntu/nbrs.p','wb') as f:
       pickle.dump(nbrs, f)

完整的追溯：

  File "/home/ubuntu/colorSetter.py", line 82, in createClassifier
    pickle.dump(nbrs, f)
  File "/usr/lib/python2.7/copy_reg.py", line 70, in _reduce_ex
    raise TypeError, "can't pickle %s objects" % base.__name__
  TypeError: can't pickle instancemethod objects

Answer 1

NearestNeighbors实例中的某个位置是一个属性，它引用您在metric参数中传递给它的实例方法。 Pickle不会挑选实例方法，因此错误。

解决这个问题的一种方法是将方法distanceCIE2000_classifier()从类中移出到常规的独立函数中（如果可能的话）。