tokmangan

Токманган

Коды для Токмангана: манипулирование токеном Генеративная состязательная сеть для генерации текста

Это модель генерации иерархической последовательности, которая сначала решает, добавлен ли бланк или нет,

Если добавлен запуск, заполните его, в противном случае решает, как использовать заданный токен (используйте, игнорируйте и т. Д.).

Инструкции

1. Запустите Tokmangan в режиме MLE

python train.py --gan_model tokmangan --mode MLE --dataset coco --unit_size 32

2. Запустите Tokmangan в режиме Gan

python train.py --gan_model tokmangan --mode GAN --dataset coco --unit_size 32

3. Сгенерировать образцы

• Для оценки

python generate_for_eval.py -g tokmangan -t GAN -d coco -s 32

• Для проверки деталей (вы можете найти файлы результатов, которые начинаются с «деталей» здесь)

python generate_for_details.py -g tokmangan -t GAN -d coco -s 32 --n_generate_per_seed 10 --gen_vd_keep_prob 0.8

• Вы также можете создать модель Maskgan, используя этот проект. (Укажите опцию в скрипте как -GAN_MODEL Maskgan или -g Maskgan )

Эксперимент по синтетическим данным

• Вы можете найти образцы текста, синтезированные здесь
• Коды для оценки размещены здесь

Требования

• Код тестируется на Tensorflow версии 1.14 для Python 3.6.
• Для оценки вам необходимо скачать внешний проект - Gansfallingshort

Ссылки

Мы очень помогли в следующих проектах.

• Маскган
• Texygen
• ГАНСФАНДИНГСХОРТ