无法腌制 Scikit 学习最近邻居分类器 - 无法腌制实例方法对象

Question

我正在尝试腌制NearestNeighbor模型，但它说不能泡菜实例方法对象。

代码：

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 不会 pickle 实例方法，因此会出现错误。

解决方法是将方法distanceCIE2000_classifier()从类中移动到常规的独立函数（如果可能）。