Понимание сигмовидного предсказания в Tensorflow? - PullRequest
0 голосов
/ 13 июля 2020

Я создаю модель тензорного потока, которая предсказывает, принадлежит ли текст к одной или нескольким категориям. Я использую сигмоидальную функцию, чтобы сделать это с бинарной кроссентропией в качестве функции потерь. У меня есть 27 различных категорий, по которым я хочу сделать прогноз. Поэтому, когда я использую модель для прогнозирования тела текста, она возвращает массив формы (len of text, 27). Таким образом, он предсказывает каждое кажущееся слово. Есть ли способ сделать так, чтобы он просто предсказывал предложение в целом? Или я мог средн. вместе эти предсказания, чтобы получить предсказание всего предложения?

...