Возможно стоит в эту сторону посмотреть
https://spacy.io/api/textcategorizerНе знаю как в этой либе но в целом похоже что если бы в таблице была категория, а для полученного текста модель сразу выдавала бы категорию, было бы гораздо ближе к тому что тебе хочется
Ну я бы лично в эту сторону глядел
Или здесь спроси
@nlp_ruИли в целом опиши задачу может быть тебе более подходящий стек предложат