├── .gitignore ├── README.md ├── data ├── ICEWS14 │ ├── entity2id.txt │ ├── relation2id.txt │ ├── stat.txt │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── ICEWS18 │ ├── entity2id.txt │ ├── relation2id.txt │ ├── stat.txt │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── WIKI │ ├── entity2id.txt │ ├── relation2id.txt │ ├── stat.txt │ ├── test.txt │ ├── train.txt │ └── valid.txt └── YAGO │ ├── entity2id.txt │ ├── relation2id.txt │ ├── stat.txt │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── eval_utils.py ├── evaluation.py ├── model_utils.py ├── openai_utils.py ├── requirements.txt ├── run_hf.py ├── run_openai.py ├── run_rule.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/usc-isi-i2/isi-tkg-icl/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/usc-isi-i2/isi-tkg-icl/HEAD/README.md -------------------------------------------------------------------------------- /data/ICEWS14/entity2id.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/usc-isi-i2/isi-tkg-icl/HEAD/data/ICEWS14/entity2id.txt -------------------------------------------------------------------------------- /data/ICEWS14/relation2id.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/usc-isi-i2/isi-tkg-icl/HEAD/data/ICEWS14/relation2id.txt -------------------------------------------------------------------------------- /data/ICEWS14/stat.txt: -------------------------------------------------------------------------------- 1 | 7128 230 0 -------------------------------------------------------------------------------- /data/ICEWS14/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/usc-isi-i2/isi-tkg-icl/HEAD/data/ICEWS14/test.txt -------------------------------------------------------------------------------- /data/ICEWS14/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/usc-isi-i2/isi-tkg-icl/HEAD/data/ICEWS14/train.txt -------------------------------------------------------------------------------- /data/ICEWS14/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/usc-isi-i2/isi-tkg-icl/HEAD/data/ICEWS14/valid.txt -------------------------------------------------------------------------------- /data/ICEWS18/entity2id.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/usc-isi-i2/isi-tkg-icl/HEAD/data/ICEWS18/entity2id.txt -------------------------------------------------------------------------------- /data/ICEWS18/relation2id.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/usc-isi-i2/isi-tkg-icl/HEAD/data/ICEWS18/relation2id.txt -------------------------------------------------------------------------------- /data/ICEWS18/stat.txt: -------------------------------------------------------------------------------- 1 | 23033 256 0 -------------------------------------------------------------------------------- /data/ICEWS18/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/usc-isi-i2/isi-tkg-icl/HEAD/data/ICEWS18/test.txt -------------------------------------------------------------------------------- /data/ICEWS18/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/usc-isi-i2/isi-tkg-icl/HEAD/data/ICEWS18/train.txt -------------------------------------------------------------------------------- /data/ICEWS18/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/usc-isi-i2/isi-tkg-icl/HEAD/data/ICEWS18/valid.txt -------------------------------------------------------------------------------- /data/WIKI/entity2id.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/usc-isi-i2/isi-tkg-icl/HEAD/data/WIKI/entity2id.txt -------------------------------------------------------------------------------- /data/WIKI/relation2id.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/usc-isi-i2/isi-tkg-icl/HEAD/data/WIKI/relation2id.txt -------------------------------------------------------------------------------- /data/WIKI/stat.txt: -------------------------------------------------------------------------------- 1 | 12554 24 232 2 | -------------------------------------------------------------------------------- /data/WIKI/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/usc-isi-i2/isi-tkg-icl/HEAD/data/WIKI/test.txt -------------------------------------------------------------------------------- /data/WIKI/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/usc-isi-i2/isi-tkg-icl/HEAD/data/WIKI/train.txt -------------------------------------------------------------------------------- /data/WIKI/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/usc-isi-i2/isi-tkg-icl/HEAD/data/WIKI/valid.txt -------------------------------------------------------------------------------- /data/YAGO/entity2id.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/usc-isi-i2/isi-tkg-icl/HEAD/data/YAGO/entity2id.txt -------------------------------------------------------------------------------- /data/YAGO/relation2id.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/usc-isi-i2/isi-tkg-icl/HEAD/data/YAGO/relation2id.txt -------------------------------------------------------------------------------- /data/YAGO/stat.txt: -------------------------------------------------------------------------------- 1 | 10623 10 189 2 | -------------------------------------------------------------------------------- /data/YAGO/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/usc-isi-i2/isi-tkg-icl/HEAD/data/YAGO/test.txt -------------------------------------------------------------------------------- /data/YAGO/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/usc-isi-i2/isi-tkg-icl/HEAD/data/YAGO/train.txt -------------------------------------------------------------------------------- /data/YAGO/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/usc-isi-i2/isi-tkg-icl/HEAD/data/YAGO/valid.txt -------------------------------------------------------------------------------- /eval_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/usc-isi-i2/isi-tkg-icl/HEAD/eval_utils.py -------------------------------------------------------------------------------- /evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/usc-isi-i2/isi-tkg-icl/HEAD/evaluation.py -------------------------------------------------------------------------------- /model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/usc-isi-i2/isi-tkg-icl/HEAD/model_utils.py -------------------------------------------------------------------------------- /openai_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/usc-isi-i2/isi-tkg-icl/HEAD/openai_utils.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | torch 2 | numpy 3 | tqdm 4 | transformers 5 | openai 6 | -------------------------------------------------------------------------------- /run_hf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/usc-isi-i2/isi-tkg-icl/HEAD/run_hf.py -------------------------------------------------------------------------------- /run_openai.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/usc-isi-i2/isi-tkg-icl/HEAD/run_openai.py -------------------------------------------------------------------------------- /run_rule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/usc-isi-i2/isi-tkg-icl/HEAD/run_rule.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/usc-isi-i2/isi-tkg-icl/HEAD/utils.py --------------------------------------------------------------------------------