Добавить несколько нейронных слоев над классификатором BERT

Имеется классификатор текстов на базе трансформера BERT.
Данная модель ИИ использует вектора BERT + knn, чтобы проклассифицировать тексты. Однако нужная точность не достигается.
Есть идея, что вместо knn использовать после BERT несколько дополнительных нейронных слоев. Так как сделано в этой работе (https://newtechaudit.ru/klassifikacziya-teksta-s-ispolzovaniem-modelej-transformerov/).
Пожалуйста, также укажите стоимость вашей работы в час,...