├── .gitignore ├── README.md ├── cascade_full.sh ├── cascade_pruned.sh ├── data ├── codex-m.zip ├── codex-s.zip ├── fb15k-237.zip ├── repodb.zip └── wn18rr.zip ├── download_models.sh ├── out ├── codex-m │ ├── t1 │ │ └── config.yaml │ ├── t1_prune │ │ └── config.yaml │ ├── t2 │ │ └── config.yaml │ └── t2_prune │ │ └── config.yaml ├── codex-s │ ├── t1 │ │ └── config.yaml │ ├── t1_prune │ │ └── config.yaml │ ├── t2 │ │ └── config.yaml │ └── t2_prune │ │ └── config.yaml ├── fb15k-237 │ ├── t1 │ │ └── config.yaml │ ├── t1_prune │ │ └── config.yaml │ └── t2_prune │ │ └── config.yaml ├── repodb │ ├── t1 │ │ └── config.yaml │ ├── t1_prune │ │ └── config.yaml │ ├── t2 │ │ └── config.yaml │ └── t2_prune │ │ └── config.yaml └── wn18rr │ ├── t1 │ └── config.yaml │ ├── t1_prune │ └── config.yaml │ └── t2_prune │ └── config.yaml ├── requirements.txt └── src ├── config.py ├── data ├── __init__.py ├── answer_selector_dataset.py ├── biencoder_dataset.py ├── crossencoder_dataset.py ├── crossencoder_reranker_dataset.py ├── reranker_dataset.py ├── score_eval_dataset.py ├── tokenize.py ├── triple_dataset.py └── util.py ├── lightning_modules ├── __init__.py ├── answer_selector.py ├── base_dense_retrieval.py ├── base_kgc_module.py ├── biencoder.py ├── crossencoder.py ├── crossencoder_reranker.py └── score_eval.py ├── main.py ├── model ├── __init__.py ├── mlp.py └── text_encoder.py ├── model_selection └── ax.py └── ranking.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/README.md -------------------------------------------------------------------------------- /cascade_full.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/cascade_full.sh -------------------------------------------------------------------------------- /cascade_pruned.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/cascade_pruned.sh -------------------------------------------------------------------------------- /data/codex-m.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/data/codex-m.zip -------------------------------------------------------------------------------- /data/codex-s.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/data/codex-s.zip -------------------------------------------------------------------------------- /data/fb15k-237.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/data/fb15k-237.zip -------------------------------------------------------------------------------- /data/repodb.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/data/repodb.zip -------------------------------------------------------------------------------- /data/wn18rr.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/data/wn18rr.zip -------------------------------------------------------------------------------- /download_models.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/download_models.sh -------------------------------------------------------------------------------- /out/codex-m/t1/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/out/codex-m/t1/config.yaml -------------------------------------------------------------------------------- /out/codex-m/t1_prune/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/out/codex-m/t1_prune/config.yaml -------------------------------------------------------------------------------- /out/codex-m/t2/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/out/codex-m/t2/config.yaml -------------------------------------------------------------------------------- /out/codex-m/t2_prune/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/out/codex-m/t2_prune/config.yaml -------------------------------------------------------------------------------- /out/codex-s/t1/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/out/codex-s/t1/config.yaml -------------------------------------------------------------------------------- /out/codex-s/t1_prune/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/out/codex-s/t1_prune/config.yaml -------------------------------------------------------------------------------- /out/codex-s/t2/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/out/codex-s/t2/config.yaml -------------------------------------------------------------------------------- /out/codex-s/t2_prune/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/out/codex-s/t2_prune/config.yaml -------------------------------------------------------------------------------- /out/fb15k-237/t1/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/out/fb15k-237/t1/config.yaml -------------------------------------------------------------------------------- /out/fb15k-237/t1_prune/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/out/fb15k-237/t1_prune/config.yaml -------------------------------------------------------------------------------- /out/fb15k-237/t2_prune/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/out/fb15k-237/t2_prune/config.yaml -------------------------------------------------------------------------------- /out/repodb/t1/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/out/repodb/t1/config.yaml -------------------------------------------------------------------------------- /out/repodb/t1_prune/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/out/repodb/t1_prune/config.yaml -------------------------------------------------------------------------------- /out/repodb/t2/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/out/repodb/t2/config.yaml -------------------------------------------------------------------------------- /out/repodb/t2_prune/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/out/repodb/t2_prune/config.yaml -------------------------------------------------------------------------------- /out/wn18rr/t1/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/out/wn18rr/t1/config.yaml -------------------------------------------------------------------------------- /out/wn18rr/t1_prune/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/out/wn18rr/t1_prune/config.yaml -------------------------------------------------------------------------------- /out/wn18rr/t2_prune/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/out/wn18rr/t2_prune/config.yaml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/requirements.txt -------------------------------------------------------------------------------- /src/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/src/config.py -------------------------------------------------------------------------------- /src/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/data/answer_selector_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/src/data/answer_selector_dataset.py -------------------------------------------------------------------------------- /src/data/biencoder_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/src/data/biencoder_dataset.py -------------------------------------------------------------------------------- /src/data/crossencoder_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/src/data/crossencoder_dataset.py -------------------------------------------------------------------------------- /src/data/crossencoder_reranker_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/src/data/crossencoder_reranker_dataset.py -------------------------------------------------------------------------------- /src/data/reranker_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/src/data/reranker_dataset.py -------------------------------------------------------------------------------- /src/data/score_eval_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/src/data/score_eval_dataset.py -------------------------------------------------------------------------------- /src/data/tokenize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/src/data/tokenize.py -------------------------------------------------------------------------------- /src/data/triple_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/src/data/triple_dataset.py -------------------------------------------------------------------------------- /src/data/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/src/data/util.py -------------------------------------------------------------------------------- /src/lightning_modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/lightning_modules/answer_selector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/src/lightning_modules/answer_selector.py -------------------------------------------------------------------------------- /src/lightning_modules/base_dense_retrieval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/src/lightning_modules/base_dense_retrieval.py -------------------------------------------------------------------------------- /src/lightning_modules/base_kgc_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/src/lightning_modules/base_kgc_module.py -------------------------------------------------------------------------------- /src/lightning_modules/biencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/src/lightning_modules/biencoder.py -------------------------------------------------------------------------------- /src/lightning_modules/crossencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/src/lightning_modules/crossencoder.py -------------------------------------------------------------------------------- /src/lightning_modules/crossencoder_reranker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/src/lightning_modules/crossencoder_reranker.py -------------------------------------------------------------------------------- /src/lightning_modules/score_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/src/lightning_modules/score_eval.py -------------------------------------------------------------------------------- /src/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/src/main.py -------------------------------------------------------------------------------- /src/model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/model/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/src/model/mlp.py -------------------------------------------------------------------------------- /src/model/text_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/src/model/text_encoder.py -------------------------------------------------------------------------------- /src/model_selection/ax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/src/model_selection/ax.py -------------------------------------------------------------------------------- /src/ranking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsafavi/cascader/HEAD/src/ranking.py --------------------------------------------------------------------------------