Есть ли способ классифицировать текст на основе некоторых заданных ключевых слов с помощью Python?

Question

Я пытался немного научиться машинному обучению для проекта, в котором я работаю. В настоящее время мне удалось классифицировать текст с помощью SVM с помощью sklearn и spacy, что дало хорошие результаты, но я хочу не только классифицироватьтекст с SVM, я также хочу классифицировать его на основе списка ключевых слов, которые у меня есть. Например: если в предложении есть слово «быстро» или «секунды», я бы хотел, чтобы оно было классифицировано как производительность.

Я действительно новичок в машинном обучении и буду очень признателен за любые советы.

Answer 1

Я предполагаю, что вы уже берете часть своих данных, классифицируете их вручную, а затем используете результат в качестве данных для обучения алгоритму SVM.

Если да, то вы можете просто добавить свой список ключевых слов (признаков) и желаемых классификаций (ярлыков) к своим данным обучения. Если вы этого еще не сделали, я бы порекомендовал использовать SnowballStemmer для ваших функций тренировочных данных.