有没有办法训练 ML 模型将单词分类？

Question

我希望训练一个 ML 模型将单词分为几个大类，在我的例子中：颜色。所以我会有一些预定义的颜色桶，例如：

let blue = ["blue", "royal", "dark blue", "light blue"]

let red = ["red", "cardinal", "dusty red", "red polka dot"]

而且我要

a) 对于模型对桶中已经存在的颜色进行分类，即如果给定“蓝色”，它将知道“蓝色”在blue桶中。

b) 模型采用以前未见过的词，例如“褪色的蓝色”，并将它们分类到正确的桶中，在这种情况下，blue基于某种置信度得分。

我不确定这是否可行，我目前的方法是一系列用于分类的 if 语句，但我想知道是否有更直观的方法来使用 ML 模型来做到这一点。