pickle.load: ImportError: No module named k_means_

Question

我正在使用 pickle 转储 sklearn.cluster.KMeans 对象，如下所示：

kmeans = KMeans(n_clusters=7)

kmeans.fit(X)

pickle.dump(kmeans, open(model_fname, "w"), protocol=2)

但是，如果我尝试重新加载此 pickle 文件：

if os.path.exists(model_fname):
    print "Loading existing model .."
    return pickle.load(open(model_fname, "rb"))

我得到：

  File "C:\Python27\lib\pickle.py", line 1130, in find_class
    __import__(module)
ImportError: No module named k_means_

如何加载此文件？

Answer 1

我刚刚将 pickle 替换为 joblib:

from sklearn.externals import joblib

python