tokmangan

Tokmangan

Kode untuk Tokmangan: Token Manipulasi Jaringan permusuhan generatif untuk pembuatan teks

Ini adalah model generasi urutan hierarkis, yang pertama kali memutuskan apakah kosong ditambahkan atau tidak,

Jika kosong ditambahkan maka isi, sebaliknya memutuskan bagaimana memanfaatkan token yang diberikan (gunakan, abaikan, dll).

Instruksi

1. Jalankan Tokmangan dalam mode MLE

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

2. Jalankan Tokmangan dalam mode GAN

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

3. Hasilkan sampel

• Untuk evaluasi

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

• Untuk memeriksa detail (Anda dapat menemukan file hasil yang dimulai dengan 'detail' di sini)

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

• Anda juga dapat menghasilkan model maskgan menggunakan pada proyek ini. (Tentukan opsi dalam skrip sebagai --gan_model maskgan atau -g maskgan )

Eksperimen Data Sintetis

• Anda dapat menemukan sampel teks yang disintesis di sini
• Kode untuk evaluasi ditempatkan di sini

Persyaratan

• Kode diuji pada TensorFlow versi 1.14 untuk Python 3.6.
• Untuk evaluasi Anda perlu mengunduh proyek eksternal - Gansfallingshort

Referensi

Kami telah banyak membantu dalam proyek -proyek berikut.

• Maskgan
• Texygen
• Gansfallingshort