├── README.md ├── batcher └── persona_batcher.py ├── dataset └── persona_dataset.py ├── decode.py ├── distill.py ├── dual.py ├── metric.py ├── model ├── bert_transformer.py └── reimpl_gpt2v0.py ├── pretrain_dual.py ├── pretrain_vae.py ├── requirements.txt ├── vaek.py └── vaep.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TingchenFu/PersonaKGC/HEAD/README.md -------------------------------------------------------------------------------- /batcher/persona_batcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TingchenFu/PersonaKGC/HEAD/batcher/persona_batcher.py -------------------------------------------------------------------------------- /dataset/persona_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TingchenFu/PersonaKGC/HEAD/dataset/persona_dataset.py -------------------------------------------------------------------------------- /decode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TingchenFu/PersonaKGC/HEAD/decode.py -------------------------------------------------------------------------------- /distill.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TingchenFu/PersonaKGC/HEAD/distill.py -------------------------------------------------------------------------------- /dual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TingchenFu/PersonaKGC/HEAD/dual.py -------------------------------------------------------------------------------- /metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TingchenFu/PersonaKGC/HEAD/metric.py -------------------------------------------------------------------------------- /model/bert_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TingchenFu/PersonaKGC/HEAD/model/bert_transformer.py -------------------------------------------------------------------------------- /model/reimpl_gpt2v0.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TingchenFu/PersonaKGC/HEAD/model/reimpl_gpt2v0.py -------------------------------------------------------------------------------- /pretrain_dual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TingchenFu/PersonaKGC/HEAD/pretrain_dual.py -------------------------------------------------------------------------------- /pretrain_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TingchenFu/PersonaKGC/HEAD/pretrain_vae.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | torch==1.7.0 2 | transformers==4.15.0 -------------------------------------------------------------------------------- /vaek.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TingchenFu/PersonaKGC/HEAD/vaek.py -------------------------------------------------------------------------------- /vaep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TingchenFu/PersonaKGC/HEAD/vaep.py --------------------------------------------------------------------------------