如何保存 scikit-learn k-means 聚类模型？

Question

目前 K-means CLustring 代码在一个方法中是这样写的：

def predict(image_path):
image = cv2.imread(image_path)
image = image.reshape((image.shape[0] * image.shape[1], 3))
clt = KMeans(n_clusters = 3,  random_state=2, n_jobs=1)
clt.fit(image)

如何将它保存到模型中，以便我可以将其转换为 Core-ML 并在我的应用程序中使用它？

Answer 1

节省：

pickle.dump(clt, open("save.pkl", "wb"))

加载：

clt = pickle.load(open("save.pkl", "rb"))

Answer 2

Core ML 目前不支持 K-means 聚类。

您可以在Swift Algorithm Club找到 K-means 的简单实现。