首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >使用kmodes将集群成员资格分配给新数据

使用kmodes将集群成员资格分配给新数据
EN

Stack Overflow用户
提问于 2022-02-10 15:04:03
回答 1查看 108关注 0票数 1

这里查看这段代码

代码语言:javascript
复制
import numpy as np
from kmodes.kmodes import KModes

# random categorical data
data = np.random.choice(20, (100, 10))

km = KModes(n_clusters=4, init='Huang', n_init=5, verbose=1)
clusters = km.fit_predict(data)

# Print the cluster centroids
print(km.cluster_centroids_)

有没有人碰巧知道如何保存“集群模型”并将其应用于新的数据?或者换句话说,集群以前看不见的数据?谢谢。

EN

回答 1

Stack Overflow用户

回答已采纳

发布于 2022-02-10 16:04:40

您可以在此任务中使用pickle

代码语言:javascript
复制
import pickle

with open('cluster_model.pickle', 'wb') as n:
    pickle.dump(km, n)

当您想要在新数据上使用它时,只需:

代码语言:javascript
复制
with open('cluster_model.pickle', 'rb') as f:
    km = pickle.load(f)

# If your new data is called "new_data", you can:
new_clusters = km.predict(new_data)
票数 2
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/71067481

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档