├── .gitignore ├── README.md ├── code ├── README.md ├── doc.md ├── evaluate_model.py ├── extra │ ├── stats.txt │ ├── vocab.txt │ └── vocab_more.txt ├── generate_doc.py ├── get_args.py ├── kenlm_scorer.py ├── modules │ ├── AttentionDecoder.py │ ├── ControllableDataset.py │ ├── ControllableModel.py │ ├── PretrainedEmbeddings.py │ ├── ScoreModelDataset.py │ ├── Vocab.py │ ├── __init__.py │ ├── beam_search.py │ ├── evaluate.py │ ├── loss.py │ ├── masked_cross_entropy.py │ ├── reward_loss.py │ ├── rewards_helper.py │ └── trainer.py ├── opts.py ├── score_model.py ├── scorer.py ├── simmat.dat ├── syn.dat ├── textual_rewards │ ├── __init__.py │ ├── resources │ │ ├── all.txt │ │ ├── awl.txt │ │ ├── awl_old.txt │ │ └── syn.dat │ └── reward_polite.py ├── train.py └── vectorizer.py └── dataset ├── test.input ├── train.input └── valid.input /.gitignore: -------------------------------------------------------------------------------- 1 | *.pyc 2 | .vscode 3 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parajain/uctf/HEAD/README.md -------------------------------------------------------------------------------- /code/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parajain/uctf/HEAD/code/README.md -------------------------------------------------------------------------------- /code/doc.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parajain/uctf/HEAD/code/doc.md -------------------------------------------------------------------------------- /code/evaluate_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parajain/uctf/HEAD/code/evaluate_model.py -------------------------------------------------------------------------------- /code/extra/stats.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parajain/uctf/HEAD/code/extra/stats.txt -------------------------------------------------------------------------------- /code/extra/vocab.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parajain/uctf/HEAD/code/extra/vocab.txt -------------------------------------------------------------------------------- /code/extra/vocab_more.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parajain/uctf/HEAD/code/extra/vocab_more.txt -------------------------------------------------------------------------------- /code/generate_doc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parajain/uctf/HEAD/code/generate_doc.py -------------------------------------------------------------------------------- /code/get_args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parajain/uctf/HEAD/code/get_args.py -------------------------------------------------------------------------------- /code/kenlm_scorer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parajain/uctf/HEAD/code/kenlm_scorer.py -------------------------------------------------------------------------------- /code/modules/AttentionDecoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parajain/uctf/HEAD/code/modules/AttentionDecoder.py -------------------------------------------------------------------------------- /code/modules/ControllableDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parajain/uctf/HEAD/code/modules/ControllableDataset.py -------------------------------------------------------------------------------- /code/modules/ControllableModel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parajain/uctf/HEAD/code/modules/ControllableModel.py -------------------------------------------------------------------------------- /code/modules/PretrainedEmbeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parajain/uctf/HEAD/code/modules/PretrainedEmbeddings.py -------------------------------------------------------------------------------- /code/modules/ScoreModelDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parajain/uctf/HEAD/code/modules/ScoreModelDataset.py -------------------------------------------------------------------------------- /code/modules/Vocab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parajain/uctf/HEAD/code/modules/Vocab.py -------------------------------------------------------------------------------- /code/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parajain/uctf/HEAD/code/modules/__init__.py -------------------------------------------------------------------------------- /code/modules/beam_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parajain/uctf/HEAD/code/modules/beam_search.py -------------------------------------------------------------------------------- /code/modules/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parajain/uctf/HEAD/code/modules/evaluate.py -------------------------------------------------------------------------------- /code/modules/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parajain/uctf/HEAD/code/modules/loss.py -------------------------------------------------------------------------------- /code/modules/masked_cross_entropy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parajain/uctf/HEAD/code/modules/masked_cross_entropy.py -------------------------------------------------------------------------------- /code/modules/reward_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parajain/uctf/HEAD/code/modules/reward_loss.py -------------------------------------------------------------------------------- /code/modules/rewards_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parajain/uctf/HEAD/code/modules/rewards_helper.py -------------------------------------------------------------------------------- /code/modules/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parajain/uctf/HEAD/code/modules/trainer.py -------------------------------------------------------------------------------- /code/opts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parajain/uctf/HEAD/code/opts.py -------------------------------------------------------------------------------- /code/score_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parajain/uctf/HEAD/code/score_model.py -------------------------------------------------------------------------------- /code/scorer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parajain/uctf/HEAD/code/scorer.py -------------------------------------------------------------------------------- /code/simmat.dat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parajain/uctf/HEAD/code/simmat.dat -------------------------------------------------------------------------------- /code/syn.dat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parajain/uctf/HEAD/code/syn.dat -------------------------------------------------------------------------------- /code/textual_rewards/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parajain/uctf/HEAD/code/textual_rewards/__init__.py -------------------------------------------------------------------------------- /code/textual_rewards/resources/all.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parajain/uctf/HEAD/code/textual_rewards/resources/all.txt -------------------------------------------------------------------------------- /code/textual_rewards/resources/awl.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parajain/uctf/HEAD/code/textual_rewards/resources/awl.txt -------------------------------------------------------------------------------- /code/textual_rewards/resources/awl_old.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parajain/uctf/HEAD/code/textual_rewards/resources/awl_old.txt -------------------------------------------------------------------------------- /code/textual_rewards/resources/syn.dat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parajain/uctf/HEAD/code/textual_rewards/resources/syn.dat -------------------------------------------------------------------------------- /code/textual_rewards/reward_polite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parajain/uctf/HEAD/code/textual_rewards/reward_polite.py -------------------------------------------------------------------------------- /code/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parajain/uctf/HEAD/code/train.py -------------------------------------------------------------------------------- /code/vectorizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parajain/uctf/HEAD/code/vectorizer.py -------------------------------------------------------------------------------- /dataset/test.input: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parajain/uctf/HEAD/dataset/test.input -------------------------------------------------------------------------------- /dataset/train.input: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parajain/uctf/HEAD/dataset/train.input -------------------------------------------------------------------------------- /dataset/valid.input: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parajain/uctf/HEAD/dataset/valid.input --------------------------------------------------------------------------------