├── .gitignore ├── README.md ├── data ├── test.json └── train.json ├── examples ├── evaluation.ipynb ├── evaluation.py ├── evaluation.sh ├── knn_classifier.pkl ├── train_knn.py └── train_knn_classifier.ipynb ├── models └── README.md ├── requirements.txt ├── rl_training ├── rl_utils.py ├── train.py ├── train_rl.sh └── training_config.yaml ├── setup.py └── src └── EntFA ├── model.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcao516/EntFA/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcao516/EntFA/HEAD/README.md -------------------------------------------------------------------------------- /data/test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcao516/EntFA/HEAD/data/test.json -------------------------------------------------------------------------------- /data/train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcao516/EntFA/HEAD/data/train.json -------------------------------------------------------------------------------- /examples/evaluation.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcao516/EntFA/HEAD/examples/evaluation.ipynb -------------------------------------------------------------------------------- /examples/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcao516/EntFA/HEAD/examples/evaluation.py -------------------------------------------------------------------------------- /examples/evaluation.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcao516/EntFA/HEAD/examples/evaluation.sh -------------------------------------------------------------------------------- /examples/knn_classifier.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcao516/EntFA/HEAD/examples/knn_classifier.pkl -------------------------------------------------------------------------------- /examples/train_knn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcao516/EntFA/HEAD/examples/train_knn.py -------------------------------------------------------------------------------- /examples/train_knn_classifier.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcao516/EntFA/HEAD/examples/train_knn_classifier.ipynb -------------------------------------------------------------------------------- /models/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcao516/EntFA/HEAD/models/README.md -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | torch>=1.7.1 2 | fairseq==0.10.2 3 | transformers>=4.17.0 4 | spacy 5 | -------------------------------------------------------------------------------- /rl_training/rl_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcao516/EntFA/HEAD/rl_training/rl_utils.py -------------------------------------------------------------------------------- /rl_training/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcao516/EntFA/HEAD/rl_training/train.py -------------------------------------------------------------------------------- /rl_training/train_rl.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcao516/EntFA/HEAD/rl_training/train_rl.sh -------------------------------------------------------------------------------- /rl_training/training_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcao516/EntFA/HEAD/rl_training/training_config.yaml -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcao516/EntFA/HEAD/setup.py -------------------------------------------------------------------------------- /src/EntFA/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcao516/EntFA/HEAD/src/EntFA/model.py -------------------------------------------------------------------------------- /src/EntFA/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcao516/EntFA/HEAD/src/EntFA/utils.py --------------------------------------------------------------------------------