Отдельная текстовая классификация по категориям с использованием наивного байесовского классификатора

Question

Я пытаюсь классифицировать набор слов в разные категории, используя наивный байесовский классификатор. Например, если есть три категории (спорт, музыка, танцы), а входные слова - Gymnastic, Guitar, ballroom, Basketball и Salsa, результат должен быть таким:

Спорт

• Баскетбол
• Гимнастика

Музыка

• гитара

Танцы

• бальные
• Salsa

Можно ли это сделать с помощью наивного байесовского классификатора?