和pickle.load：导入错误：没有模块名为k_means_

Question

我倾倒使用pickle这样的sklearn.cluster.KMeans对象：

kmeans = KMeans(n_clusters=7) 

kmeans.fit(X) 

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

但是，如果我尝试重新加载该文件泡菜：

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

我只是joblib取代pickle：

from sklearn.externals import joblib