A reproduction to GraphWriter (《Text Generation from Knowledge Graphs with Graph Transformers》).
- train:
python train.py --save=[save dir] --gpus=[gpus]
- generate:
python generate.py --save=[save dir] --generate_from=[pkl file name] --generated_file=[target file]
- get ground truth:
python gene_ref.py
- eval:
python eval.py [generated file] [ground truth]
python train.py --save=./save/test
python generate.py --save=./save/test --generate_from=epoch_10.pkl
python gene_ref.py
python eval.py ./save/test/gene.txt test_ref.txt
Implementation | Bleu | METROR |
---|---|---|
Original | 14.3 | 18.8 |
My(w/o transformer) | 12.5 | 17.5 |
My(w transformer) | 12.2 | - |
My(w transformer,+ glove) | 12.3 | - |
好吧我还没有完全复现出他的结果(ノДT)