Есть ли способ управления длиной вывода последовательности для модели суммирования текста последовательности?

Есть ли способ контролировать количество слов или символов, которые производит модель seq2seq для суммирования текста? Примеры:

"Моя собака - самая быстрая собака в мире. Он тоже любит обниматься".

1 вывод: Моя собака быстрая и любит обниматься.

2 вывод: моя собака самая быстрая собака и тоже любит обниматься.

0 ответов

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

Надеюсь, это будет полезно

Другие вопросы по тегам