Корни слов ищутся по электронному словарю Зализняка. Он для этого и существует.
Как-то давно выкладывал. У меня точно не осталось.
Вот, что может помочь отцу русской демократии: https://yadi.sk/d/YW7pOk3QpcUgxw Нейросети для других задач.
+ к тому-же, как я понял, понадобится переменное число входов(длина слова-предложения).
Это не простая задача. Надо поискать, как то делал, но не обещаю, много чего безвозвратно погибло у меня.
https://habr.com/ru/company/meanotek/blog/256593/
Но всё равно, замучаешься обучать, если обратной ошибкой, да и другими методами тоже.
Есть правило - Если можно обойтись без сети, - делай без сети!
Результат будет таким: - "Это скорее белое, чем чёрное, с вероятностью N%".

Что надо-то? В тексте надо заменять, или кто?
Можешь на пальцах показать?
P.S.
Кое-чего нашёл. Схемка ищет однокоренные слова, понимает смысл предложения, в следствии чего точно определяет [все-всё, гОры-горЫ, зАмок-замОк, бОльшая-большАя, бОчки-бочкИ...], понимает смысл прочитанного рассказа\книги и делает краткое описание - аннотацию.
Может то, что тебе нужно.
Только надо схемку привести в порядок.
Ты сначала объясни, что нужно, а то я зря буду время тратить.