добавлен плохо работающий подсчет вероятности каждой категории