我希望訓練一個 ML 模型將單詞分為幾個大類,在我的例子中:顏色。所以我會有一些預定義的顏色桶,例如:let blue = ["blue", "royal", "dark blue", "light blue"]let red = ["red", "cardinal", "dusty red", "red polka dot"]而且我要a) 對于模型對桶中已經存在的顏色進行分類,即如果給定“藍色”,它將知道“藍色”在blue桶中。b) 模型采用以前未見過的詞,例如“褪色的藍色”,并將它們分類到正確的桶中,在這種情況下,blue基于某種置信度得分。我不確定這是否可行,我目前的方法是一系列用于分類的 if 語句,但我想知道是否有更直觀的方法來使用 ML 模型來做到這一點。
添加回答
舉報
0/150
提交
取消