【问题标题】：How can I sort words into categories in Python?如何在 Python 中将单词分类？
【发布时间】：2023-04-05 23:18:01
【问题描述】：

我从事一个项目，我使用 Google 视觉来检测图像中的对象。 API 返回标签列表。所以我有多个单词，我想将每个单词归入一个类别。例如：

谷歌云视觉回归：

['Head', 'Lamp', 'Eye', 'Green', 'Arm', 'Piano', 'Mobile phone', 'Blue', 'Toy']

我想要类似的东西：

{'Object' : ['Lamp', 'Piano', 'Mobile phone', 'Toy'],
'Color' : ['Green', 'Blue'],
'Body parts': ['Head', 'Eye', 'Arm']
}

我知道 word2vec 有一种叫做相似性的东西，但这意味着我必须训练一个模型。有没有我可以使用的预置模型？或者也许是另一种解决方案？

【问题讨论】：

    标签：
    python
    machine-learning
    nlp
    word2vec
    similarity