├── README.md ├── data ├── ecbd │ ├── ecbd_2022_1k_augmented.json │ ├── ecbd_2023_augmented.json │ └── specificity_popular_20np_20random.json └── entity_inferences │ └── entity_inferences_augmented.json ├── experiments ├── gpt_ecbd.py └── gpt_entity_inferences.py ├── requirements.txt └── src ├── data_utils.py ├── distill_gpt.py ├── edit_func.py ├── metrics.py ├── run_edit.py └── trainer.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shankarp8/knowledge_distillation/HEAD/README.md -------------------------------------------------------------------------------- /data/ecbd/ecbd_2022_1k_augmented.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shankarp8/knowledge_distillation/HEAD/data/ecbd/ecbd_2022_1k_augmented.json -------------------------------------------------------------------------------- /data/ecbd/ecbd_2023_augmented.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shankarp8/knowledge_distillation/HEAD/data/ecbd/ecbd_2023_augmented.json -------------------------------------------------------------------------------- /data/ecbd/specificity_popular_20np_20random.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shankarp8/knowledge_distillation/HEAD/data/ecbd/specificity_popular_20np_20random.json -------------------------------------------------------------------------------- /data/entity_inferences/entity_inferences_augmented.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shankarp8/knowledge_distillation/HEAD/data/entity_inferences/entity_inferences_augmented.json -------------------------------------------------------------------------------- /experiments/gpt_ecbd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shankarp8/knowledge_distillation/HEAD/experiments/gpt_ecbd.py -------------------------------------------------------------------------------- /experiments/gpt_entity_inferences.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shankarp8/knowledge_distillation/HEAD/experiments/gpt_entity_inferences.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shankarp8/knowledge_distillation/HEAD/requirements.txt -------------------------------------------------------------------------------- /src/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shankarp8/knowledge_distillation/HEAD/src/data_utils.py -------------------------------------------------------------------------------- /src/distill_gpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shankarp8/knowledge_distillation/HEAD/src/distill_gpt.py -------------------------------------------------------------------------------- /src/edit_func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shankarp8/knowledge_distillation/HEAD/src/edit_func.py -------------------------------------------------------------------------------- /src/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shankarp8/knowledge_distillation/HEAD/src/metrics.py -------------------------------------------------------------------------------- /src/run_edit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shankarp8/knowledge_distillation/HEAD/src/run_edit.py -------------------------------------------------------------------------------- /src/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shankarp8/knowledge_distillation/HEAD/src/trainer.py --------------------------------------------------------------------------------