├── .gitignore ├── LICENSE ├── README.md ├── __init__.py ├── anserini_dependency ├── README.md ├── RetrieveSentences.py ├── api.py └── js │ ├── Icon.icns │ ├── IconTemplate.png │ ├── IconTemplate@2x.png │ ├── index.html │ ├── index.js │ ├── package.json │ ├── recorder.js │ ├── recorderWorker.js │ └── speech.js ├── common ├── __init__.py ├── dataset.py ├── evaluation.py ├── evaluators │ ├── __init__.py │ ├── evaluator.py │ ├── msrvid_evaluator.py │ ├── pit2015_evaluator.py │ ├── qa_evaluator.py │ ├── quora_evaluator.py │ ├── sick_evaluator.py │ ├── snli_evaluator.py │ ├── sst_evaluator.py │ ├── sts2014_evaluator.py │ ├── trecqa_evaluator.py │ └── wikiqa_evaluator.py ├── train.py └── trainers │ ├── __init__.py │ ├── msrvid_trainer.py │ ├── pit2015_trainer.py │ ├── qa_trainer.py │ ├── quora_trainer.py │ ├── sick_trainer.py │ ├── snli_trainer.py │ ├── sst_trainer.py │ ├── sts2014_trainer.py │ ├── trainer.py │ ├── trecqa_trainer.py │ └── wikiqa_trainer.py ├── config.cfg.example ├── conv_rnn ├── README.md ├── data.py ├── getData.sh ├── model.py ├── test.py └── train.py ├── datasets ├── __init__.py ├── castor_dataset.py ├── idf_utils.py ├── msrvid.py ├── pit2015.py ├── quora.py ├── sick.py ├── snli.py ├── sst.py ├── sts2014.py ├── trecqa.py └── wikiqa.py ├── decatt ├── README.md ├── __init__.py ├── __main__.py └── model.py ├── docs ├── Using SHARCNET.ipynb └── internal-instructions.md ├── esim ├── README.md ├── __init__.py ├── __main__.py └── model.py ├── idf_baseline ├── README.md ├── experimental_settings.py ├── pom.xml ├── qa-data-only-idf.py └── src │ └── main │ ├── java │ └── ai │ │ └── castor │ │ └── idf │ │ ├── FetchTermIDF.java │ │ └── IDFScorer.java │ └── resources │ └── ai │ └── castor │ └── qa │ └── english-stoplist.txt ├── mp_cnn ├── README.md ├── __init__.py ├── __main__.py ├── lite_model.py └── model.py ├── nce ├── nce_pairwise_mp │ ├── README.md │ ├── __init__.py │ ├── evaluators │ │ ├── __init__.py │ │ ├── qa_evaluator.py │ │ ├── trecqa_evaluator.py │ │ └── wikiqa_evaluator.py │ ├── main.py │ ├── train_script.py │ └── trainers │ │ ├── __init__.py │ │ ├── qa_trainer.py │ │ ├── trecqa_trainer.py │ │ └── wikiqa_trainer.py └── nce_pairwise_sm │ ├── README.md │ ├── args.py │ ├── main.py │ ├── model.py │ ├── overlap_features.py │ ├── requirements.txt │ └── train.py ├── requirements.txt ├── run_ui.sh ├── setup.py ├── sm_cnn ├── .gitignore ├── README.md ├── __init.py__.py ├── args.py ├── bridge.py ├── create_dataset.sh ├── external_features.py ├── main.py ├── model.py ├── overlap_features.py ├── train.py ├── trec_dataset.py └── wiki_dataset.py ├── sse ├── README.md ├── __init__.py ├── __main__.py └── model.py ├── utils ├── __init__.py ├── build_w2v.py ├── get_trec_eval.sh ├── nce_neighbors.py ├── relevancy_metrics.py ├── serialization.py └── torch_util.py └── vdpwi ├── README.md ├── __main__.py ├── data.py ├── model.py ├── train_default.sh └── utils ├── __init__.py ├── log.py ├── preprocess.py └── tune.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /anserini_dependency/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/anserini_dependency/README.md -------------------------------------------------------------------------------- /anserini_dependency/RetrieveSentences.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/anserini_dependency/RetrieveSentences.py -------------------------------------------------------------------------------- /anserini_dependency/api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/anserini_dependency/api.py -------------------------------------------------------------------------------- /anserini_dependency/js/Icon.icns: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/anserini_dependency/js/Icon.icns -------------------------------------------------------------------------------- /anserini_dependency/js/IconTemplate.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/anserini_dependency/js/IconTemplate.png -------------------------------------------------------------------------------- /anserini_dependency/js/IconTemplate@2x.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/anserini_dependency/js/IconTemplate@2x.png -------------------------------------------------------------------------------- /anserini_dependency/js/index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/anserini_dependency/js/index.html -------------------------------------------------------------------------------- /anserini_dependency/js/index.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/anserini_dependency/js/index.js -------------------------------------------------------------------------------- /anserini_dependency/js/package.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/anserini_dependency/js/package.json -------------------------------------------------------------------------------- /anserini_dependency/js/recorder.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/anserini_dependency/js/recorder.js -------------------------------------------------------------------------------- /anserini_dependency/js/recorderWorker.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/anserini_dependency/js/recorderWorker.js -------------------------------------------------------------------------------- /anserini_dependency/js/speech.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/anserini_dependency/js/speech.js -------------------------------------------------------------------------------- /common/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/common/__init__.py -------------------------------------------------------------------------------- /common/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/common/dataset.py -------------------------------------------------------------------------------- /common/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/common/evaluation.py -------------------------------------------------------------------------------- /common/evaluators/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /common/evaluators/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/common/evaluators/evaluator.py -------------------------------------------------------------------------------- /common/evaluators/msrvid_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/common/evaluators/msrvid_evaluator.py -------------------------------------------------------------------------------- /common/evaluators/pit2015_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/common/evaluators/pit2015_evaluator.py -------------------------------------------------------------------------------- /common/evaluators/qa_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/common/evaluators/qa_evaluator.py -------------------------------------------------------------------------------- /common/evaluators/quora_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/common/evaluators/quora_evaluator.py -------------------------------------------------------------------------------- /common/evaluators/sick_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/common/evaluators/sick_evaluator.py -------------------------------------------------------------------------------- /common/evaluators/snli_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/common/evaluators/snli_evaluator.py -------------------------------------------------------------------------------- /common/evaluators/sst_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/common/evaluators/sst_evaluator.py -------------------------------------------------------------------------------- /common/evaluators/sts2014_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/common/evaluators/sts2014_evaluator.py -------------------------------------------------------------------------------- /common/evaluators/trecqa_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/common/evaluators/trecqa_evaluator.py -------------------------------------------------------------------------------- /common/evaluators/wikiqa_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/common/evaluators/wikiqa_evaluator.py -------------------------------------------------------------------------------- /common/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/common/train.py -------------------------------------------------------------------------------- /common/trainers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /common/trainers/msrvid_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/common/trainers/msrvid_trainer.py -------------------------------------------------------------------------------- /common/trainers/pit2015_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/common/trainers/pit2015_trainer.py -------------------------------------------------------------------------------- /common/trainers/qa_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/common/trainers/qa_trainer.py -------------------------------------------------------------------------------- /common/trainers/quora_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/common/trainers/quora_trainer.py -------------------------------------------------------------------------------- /common/trainers/sick_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/common/trainers/sick_trainer.py -------------------------------------------------------------------------------- /common/trainers/snli_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/common/trainers/snli_trainer.py -------------------------------------------------------------------------------- /common/trainers/sst_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/common/trainers/sst_trainer.py -------------------------------------------------------------------------------- /common/trainers/sts2014_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/common/trainers/sts2014_trainer.py -------------------------------------------------------------------------------- /common/trainers/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/common/trainers/trainer.py -------------------------------------------------------------------------------- /common/trainers/trecqa_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/common/trainers/trecqa_trainer.py -------------------------------------------------------------------------------- /common/trainers/wikiqa_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/common/trainers/wikiqa_trainer.py -------------------------------------------------------------------------------- /config.cfg.example: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/config.cfg.example -------------------------------------------------------------------------------- /conv_rnn/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/conv_rnn/README.md -------------------------------------------------------------------------------- /conv_rnn/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/conv_rnn/data.py -------------------------------------------------------------------------------- /conv_rnn/getData.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/conv_rnn/getData.sh -------------------------------------------------------------------------------- /conv_rnn/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/conv_rnn/model.py -------------------------------------------------------------------------------- /conv_rnn/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/conv_rnn/test.py -------------------------------------------------------------------------------- /conv_rnn/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/conv_rnn/train.py -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /datasets/castor_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/datasets/castor_dataset.py -------------------------------------------------------------------------------- /datasets/idf_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/datasets/idf_utils.py -------------------------------------------------------------------------------- /datasets/msrvid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/datasets/msrvid.py -------------------------------------------------------------------------------- /datasets/pit2015.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/datasets/pit2015.py -------------------------------------------------------------------------------- /datasets/quora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/datasets/quora.py -------------------------------------------------------------------------------- /datasets/sick.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/datasets/sick.py -------------------------------------------------------------------------------- /datasets/snli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/datasets/snli.py -------------------------------------------------------------------------------- /datasets/sst.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/datasets/sst.py -------------------------------------------------------------------------------- /datasets/sts2014.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/datasets/sts2014.py -------------------------------------------------------------------------------- /datasets/trecqa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/datasets/trecqa.py -------------------------------------------------------------------------------- /datasets/wikiqa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/datasets/wikiqa.py -------------------------------------------------------------------------------- /decatt/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/decatt/README.md -------------------------------------------------------------------------------- /decatt/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /decatt/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/decatt/__main__.py -------------------------------------------------------------------------------- /decatt/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/decatt/model.py -------------------------------------------------------------------------------- /docs/Using SHARCNET.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/docs/Using SHARCNET.ipynb -------------------------------------------------------------------------------- /docs/internal-instructions.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/docs/internal-instructions.md -------------------------------------------------------------------------------- /esim/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/esim/README.md -------------------------------------------------------------------------------- /esim/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /esim/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/esim/__main__.py -------------------------------------------------------------------------------- /esim/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/esim/model.py -------------------------------------------------------------------------------- /idf_baseline/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/idf_baseline/README.md -------------------------------------------------------------------------------- /idf_baseline/experimental_settings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/idf_baseline/experimental_settings.py -------------------------------------------------------------------------------- /idf_baseline/pom.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/idf_baseline/pom.xml -------------------------------------------------------------------------------- /idf_baseline/qa-data-only-idf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/idf_baseline/qa-data-only-idf.py -------------------------------------------------------------------------------- /idf_baseline/src/main/java/ai/castor/idf/FetchTermIDF.java: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/idf_baseline/src/main/java/ai/castor/idf/FetchTermIDF.java -------------------------------------------------------------------------------- /idf_baseline/src/main/java/ai/castor/idf/IDFScorer.java: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/idf_baseline/src/main/java/ai/castor/idf/IDFScorer.java -------------------------------------------------------------------------------- /idf_baseline/src/main/resources/ai/castor/qa/english-stoplist.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/idf_baseline/src/main/resources/ai/castor/qa/english-stoplist.txt -------------------------------------------------------------------------------- /mp_cnn/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/mp_cnn/README.md -------------------------------------------------------------------------------- /mp_cnn/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /mp_cnn/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/mp_cnn/__main__.py -------------------------------------------------------------------------------- /mp_cnn/lite_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/mp_cnn/lite_model.py -------------------------------------------------------------------------------- /mp_cnn/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/mp_cnn/model.py -------------------------------------------------------------------------------- /nce/nce_pairwise_mp/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/nce/nce_pairwise_mp/README.md -------------------------------------------------------------------------------- /nce/nce_pairwise_mp/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /nce/nce_pairwise_mp/evaluators/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /nce/nce_pairwise_mp/evaluators/qa_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/nce/nce_pairwise_mp/evaluators/qa_evaluator.py -------------------------------------------------------------------------------- /nce/nce_pairwise_mp/evaluators/trecqa_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/nce/nce_pairwise_mp/evaluators/trecqa_evaluator.py -------------------------------------------------------------------------------- /nce/nce_pairwise_mp/evaluators/wikiqa_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/nce/nce_pairwise_mp/evaluators/wikiqa_evaluator.py -------------------------------------------------------------------------------- /nce/nce_pairwise_mp/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/nce/nce_pairwise_mp/main.py -------------------------------------------------------------------------------- /nce/nce_pairwise_mp/train_script.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/nce/nce_pairwise_mp/train_script.py -------------------------------------------------------------------------------- /nce/nce_pairwise_mp/trainers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /nce/nce_pairwise_mp/trainers/qa_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/nce/nce_pairwise_mp/trainers/qa_trainer.py -------------------------------------------------------------------------------- /nce/nce_pairwise_mp/trainers/trecqa_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/nce/nce_pairwise_mp/trainers/trecqa_trainer.py -------------------------------------------------------------------------------- /nce/nce_pairwise_mp/trainers/wikiqa_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/nce/nce_pairwise_mp/trainers/wikiqa_trainer.py -------------------------------------------------------------------------------- /nce/nce_pairwise_sm/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/nce/nce_pairwise_sm/README.md -------------------------------------------------------------------------------- /nce/nce_pairwise_sm/args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/nce/nce_pairwise_sm/args.py -------------------------------------------------------------------------------- /nce/nce_pairwise_sm/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/nce/nce_pairwise_sm/main.py -------------------------------------------------------------------------------- /nce/nce_pairwise_sm/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/nce/nce_pairwise_sm/model.py -------------------------------------------------------------------------------- /nce/nce_pairwise_sm/overlap_features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/nce/nce_pairwise_sm/overlap_features.py -------------------------------------------------------------------------------- /nce/nce_pairwise_sm/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/nce/nce_pairwise_sm/requirements.txt -------------------------------------------------------------------------------- /nce/nce_pairwise_sm/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/nce/nce_pairwise_sm/train.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/requirements.txt -------------------------------------------------------------------------------- /run_ui.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/run_ui.sh -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/setup.py -------------------------------------------------------------------------------- /sm_cnn/.gitignore: -------------------------------------------------------------------------------- 1 | *pyc 2 | *.pt 3 | text/ 4 | trained_models/ 5 | data/ 6 | -------------------------------------------------------------------------------- /sm_cnn/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/sm_cnn/README.md -------------------------------------------------------------------------------- /sm_cnn/__init.py__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sm_cnn/args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/sm_cnn/args.py -------------------------------------------------------------------------------- /sm_cnn/bridge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/sm_cnn/bridge.py -------------------------------------------------------------------------------- /sm_cnn/create_dataset.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/sm_cnn/create_dataset.sh -------------------------------------------------------------------------------- /sm_cnn/external_features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/sm_cnn/external_features.py -------------------------------------------------------------------------------- /sm_cnn/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/sm_cnn/main.py -------------------------------------------------------------------------------- /sm_cnn/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/sm_cnn/model.py -------------------------------------------------------------------------------- /sm_cnn/overlap_features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/sm_cnn/overlap_features.py -------------------------------------------------------------------------------- /sm_cnn/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/sm_cnn/train.py -------------------------------------------------------------------------------- /sm_cnn/trec_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/sm_cnn/trec_dataset.py -------------------------------------------------------------------------------- /sm_cnn/wiki_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/sm_cnn/wiki_dataset.py -------------------------------------------------------------------------------- /sse/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/sse/README.md -------------------------------------------------------------------------------- /sse/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sse/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/sse/__main__.py -------------------------------------------------------------------------------- /sse/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/sse/model.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/build_w2v.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/utils/build_w2v.py -------------------------------------------------------------------------------- /utils/get_trec_eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/utils/get_trec_eval.sh -------------------------------------------------------------------------------- /utils/nce_neighbors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/utils/nce_neighbors.py -------------------------------------------------------------------------------- /utils/relevancy_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/utils/relevancy_metrics.py -------------------------------------------------------------------------------- /utils/serialization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/utils/serialization.py -------------------------------------------------------------------------------- /utils/torch_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/utils/torch_util.py -------------------------------------------------------------------------------- /vdpwi/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/vdpwi/README.md -------------------------------------------------------------------------------- /vdpwi/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/vdpwi/__main__.py -------------------------------------------------------------------------------- /vdpwi/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/vdpwi/data.py -------------------------------------------------------------------------------- /vdpwi/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/vdpwi/model.py -------------------------------------------------------------------------------- /vdpwi/train_default.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/vdpwi/train_default.sh -------------------------------------------------------------------------------- /vdpwi/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /vdpwi/utils/log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/vdpwi/utils/log.py -------------------------------------------------------------------------------- /vdpwi/utils/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/vdpwi/utils/preprocess.py -------------------------------------------------------------------------------- /vdpwi/utils/tune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castorini/castor/HEAD/vdpwi/utils/tune.py --------------------------------------------------------------------------------