首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >如何将聚类应用于句子嵌入?

如何将聚类应用于句子嵌入?
EN

Stack Overflow用户
提问于 2019-07-24 19:41:38
回答 1查看 710关注 0票数 1

我想用原始文件的要点创建一个摘要。为此,我使用通用句子编码器(https://tfhub.dev/google/universal-sentence-encoder/2)进行了句子嵌入。之后,我想对我的向量应用聚类。

我尝试过使用库sklearn

代码语言:javascript
运行
复制
import numpy as np
from sklearn.cluster import KMeans

n_clusters = np.ceil(len(encoded)**0.5)
kmeans = KMeans(n_clusters=n_clusters)
kmeans = kmeans.fit(encoded)

但是我得到了一个错误消息:

代码语言:javascript
运行
复制
'numpy.float64' object cannot be interpreted as an integer'
EN

回答 1

Stack Overflow用户

发布于 2019-10-08 21:08:21

问题是由下面这行引起的:

代码语言:javascript
运行
复制
n_clusters = np.ceil(len(encoded)**0.5)

kmeans期望收到集群数量的integer,因此只需添加:

代码语言:javascript
运行
复制
n_clusters = int(np.ceil(len(encoded)**0.5))
票数 1
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/57182093

复制
相关文章

相似问题

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