tokmangan

Tokmangan

Códigos para Tokmangan: Manipulação de token Rede adversária generativa para geração de texto

Este é um modelo hierárquico de geração de sequência, que primeiro decide se um espaço em branco é adicionado ou não,

Se um espaço em branco for adicionado, preencha -o, caso contrário, decida como utilizar um determinado token (use, ignorar, etc.).

Instruções

1. Corra Tokmangan no modo MLE

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

2. Corra Tokmangan no modo GaN

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

3. Gere amostras

• Para avaliação

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

• Para verificar detalhes (você pode encontrar arquivos de resultado que começam com 'detalhes' aqui)

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

• Você também pode produzir o modelo Maskgan usando neste projeto. (Especifique a opção no script como --gan_model Maskgan ou -g Maskgan )

Experiência de dados sintéticos

• Você pode encontrar amostras de texto sintetizadas aqui
• Os códigos de avaliação são colocados aqui

Requisitos

• O código é testado no TensorFlow versão 1.14 para Python 3.6.
• Para avaliação, você precisa baixar o projeto externo - Gansfallingshort

Referências

Ajudamos muito nos seguintes projetos.

• Maskgan
• Texygen
• Gansfallingshort