├── CODEOWNERS ├── CODE_OF_CONDUCT.md ├── LICENSE.txt ├── README.md ├── SECURITY.md ├── extractors ├── README.md ├── add_rouge.py ├── postprocess_relreg.py ├── predict_qmsum_bart.sh ├── prep_data_relreg.py ├── run_relreg.sh ├── run_relreg_tt.sh ├── test_relreg_tt.py ├── train_qmsum_bart.sh └── train_relreg_tt.py ├── multiencoder ├── README.md ├── convert_qmsum.py ├── dataset.py ├── models.py ├── report_training_runs.py ├── scripts │ ├── predict_qmsum_16_1024_strided_val.sh │ ├── predict_qmsum_16_256_strided_val.sh │ ├── predict_qmsum_16_512_strided_val.sh │ ├── predict_qmsum_32_256_strided_val.sh │ ├── predict_qmsum_32_512_nostrided_val.sh │ ├── predict_qmsum_32_512_strided_test.sh │ ├── predict_qmsum_32_512_strided_val.sh │ ├── predict_qmsum_32_512_strided_wikisum_test.sh │ ├── predict_qmsum_32_512_strided_wikisum_val.sh │ ├── predict_qmsum_4_1024_strided_val.sh │ ├── predict_qmsum_64_256_strided_val.sh │ ├── predict_qmsum_8_1024_strided_val.sh │ ├── predict_qmsum_8_512_strided_val.sh │ ├── predict_val.sh │ ├── report_rouge_test.sh │ ├── report_rouge_val.sh │ ├── select_checkpoints.sh │ ├── train_qmsum_16_1024_strided.sh │ ├── train_qmsum_16_256_strided.sh │ ├── train_qmsum_16_256_strided_catchup.sh │ ├── train_qmsum_16_512_strided.sh │ ├── train_qmsum_32_256_strided.sh │ ├── train_qmsum_32_512_nostrided.sh │ ├── train_qmsum_32_512_strided.sh │ ├── train_qmsum_32_512_strided_wikisum.sh │ ├── train_qmsum_4_1024_strided.sh │ ├── train_qmsum_64_256_strided.sh │ ├── train_qmsum_8_1024_strided.sh │ └── train_qmsum_8_512_strided.sh ├── select_checkpoints.py ├── test │ ├── __init__.py │ ├── data │ │ └── dataset.jsonl │ ├── test_dataset.py │ └── test_model.py ├── train.py └── transformers │ ├── LICENSE │ └── changes ├── preprocessing ├── README.md └── prep_qmsum.py ├── requirements.txt └── rouge ├── __init__.py └── report_rouge.py /CODEOWNERS: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/CODEOWNERS -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/README.md -------------------------------------------------------------------------------- /SECURITY.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/SECURITY.md -------------------------------------------------------------------------------- /extractors/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/extractors/README.md -------------------------------------------------------------------------------- /extractors/add_rouge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/extractors/add_rouge.py -------------------------------------------------------------------------------- /extractors/postprocess_relreg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/extractors/postprocess_relreg.py -------------------------------------------------------------------------------- /extractors/predict_qmsum_bart.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/extractors/predict_qmsum_bart.sh -------------------------------------------------------------------------------- /extractors/prep_data_relreg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/extractors/prep_data_relreg.py -------------------------------------------------------------------------------- /extractors/run_relreg.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/extractors/run_relreg.sh -------------------------------------------------------------------------------- /extractors/run_relreg_tt.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/extractors/run_relreg_tt.sh -------------------------------------------------------------------------------- /extractors/test_relreg_tt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/extractors/test_relreg_tt.py -------------------------------------------------------------------------------- /extractors/train_qmsum_bart.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/extractors/train_qmsum_bart.sh -------------------------------------------------------------------------------- /extractors/train_relreg_tt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/extractors/train_relreg_tt.py -------------------------------------------------------------------------------- /multiencoder/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/multiencoder/README.md -------------------------------------------------------------------------------- /multiencoder/convert_qmsum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/multiencoder/convert_qmsum.py -------------------------------------------------------------------------------- /multiencoder/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/multiencoder/dataset.py -------------------------------------------------------------------------------- /multiencoder/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/multiencoder/models.py -------------------------------------------------------------------------------- /multiencoder/report_training_runs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/multiencoder/report_training_runs.py -------------------------------------------------------------------------------- /multiencoder/scripts/predict_qmsum_16_1024_strided_val.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/multiencoder/scripts/predict_qmsum_16_1024_strided_val.sh -------------------------------------------------------------------------------- /multiencoder/scripts/predict_qmsum_16_256_strided_val.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/multiencoder/scripts/predict_qmsum_16_256_strided_val.sh -------------------------------------------------------------------------------- /multiencoder/scripts/predict_qmsum_16_512_strided_val.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/multiencoder/scripts/predict_qmsum_16_512_strided_val.sh -------------------------------------------------------------------------------- /multiencoder/scripts/predict_qmsum_32_256_strided_val.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/multiencoder/scripts/predict_qmsum_32_256_strided_val.sh -------------------------------------------------------------------------------- /multiencoder/scripts/predict_qmsum_32_512_nostrided_val.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/multiencoder/scripts/predict_qmsum_32_512_nostrided_val.sh -------------------------------------------------------------------------------- /multiencoder/scripts/predict_qmsum_32_512_strided_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/multiencoder/scripts/predict_qmsum_32_512_strided_test.sh -------------------------------------------------------------------------------- /multiencoder/scripts/predict_qmsum_32_512_strided_val.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/multiencoder/scripts/predict_qmsum_32_512_strided_val.sh -------------------------------------------------------------------------------- /multiencoder/scripts/predict_qmsum_32_512_strided_wikisum_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/multiencoder/scripts/predict_qmsum_32_512_strided_wikisum_test.sh -------------------------------------------------------------------------------- /multiencoder/scripts/predict_qmsum_32_512_strided_wikisum_val.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/multiencoder/scripts/predict_qmsum_32_512_strided_wikisum_val.sh -------------------------------------------------------------------------------- /multiencoder/scripts/predict_qmsum_4_1024_strided_val.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/multiencoder/scripts/predict_qmsum_4_1024_strided_val.sh -------------------------------------------------------------------------------- /multiencoder/scripts/predict_qmsum_64_256_strided_val.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/multiencoder/scripts/predict_qmsum_64_256_strided_val.sh -------------------------------------------------------------------------------- /multiencoder/scripts/predict_qmsum_8_1024_strided_val.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/multiencoder/scripts/predict_qmsum_8_1024_strided_val.sh -------------------------------------------------------------------------------- /multiencoder/scripts/predict_qmsum_8_512_strided_val.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/multiencoder/scripts/predict_qmsum_8_512_strided_val.sh -------------------------------------------------------------------------------- /multiencoder/scripts/predict_val.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/multiencoder/scripts/predict_val.sh -------------------------------------------------------------------------------- /multiencoder/scripts/report_rouge_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/multiencoder/scripts/report_rouge_test.sh -------------------------------------------------------------------------------- /multiencoder/scripts/report_rouge_val.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/multiencoder/scripts/report_rouge_val.sh -------------------------------------------------------------------------------- /multiencoder/scripts/select_checkpoints.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/multiencoder/scripts/select_checkpoints.sh -------------------------------------------------------------------------------- /multiencoder/scripts/train_qmsum_16_1024_strided.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/multiencoder/scripts/train_qmsum_16_1024_strided.sh -------------------------------------------------------------------------------- /multiencoder/scripts/train_qmsum_16_256_strided.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/multiencoder/scripts/train_qmsum_16_256_strided.sh -------------------------------------------------------------------------------- /multiencoder/scripts/train_qmsum_16_256_strided_catchup.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/multiencoder/scripts/train_qmsum_16_256_strided_catchup.sh -------------------------------------------------------------------------------- /multiencoder/scripts/train_qmsum_16_512_strided.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/multiencoder/scripts/train_qmsum_16_512_strided.sh -------------------------------------------------------------------------------- /multiencoder/scripts/train_qmsum_32_256_strided.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/multiencoder/scripts/train_qmsum_32_256_strided.sh -------------------------------------------------------------------------------- /multiencoder/scripts/train_qmsum_32_512_nostrided.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/multiencoder/scripts/train_qmsum_32_512_nostrided.sh -------------------------------------------------------------------------------- /multiencoder/scripts/train_qmsum_32_512_strided.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/multiencoder/scripts/train_qmsum_32_512_strided.sh -------------------------------------------------------------------------------- /multiencoder/scripts/train_qmsum_32_512_strided_wikisum.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/multiencoder/scripts/train_qmsum_32_512_strided_wikisum.sh -------------------------------------------------------------------------------- /multiencoder/scripts/train_qmsum_4_1024_strided.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/multiencoder/scripts/train_qmsum_4_1024_strided.sh -------------------------------------------------------------------------------- /multiencoder/scripts/train_qmsum_64_256_strided.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/multiencoder/scripts/train_qmsum_64_256_strided.sh -------------------------------------------------------------------------------- /multiencoder/scripts/train_qmsum_8_1024_strided.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/multiencoder/scripts/train_qmsum_8_1024_strided.sh -------------------------------------------------------------------------------- /multiencoder/scripts/train_qmsum_8_512_strided.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/multiencoder/scripts/train_qmsum_8_512_strided.sh -------------------------------------------------------------------------------- /multiencoder/select_checkpoints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/multiencoder/select_checkpoints.py -------------------------------------------------------------------------------- /multiencoder/test/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /multiencoder/test/data/dataset.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/multiencoder/test/data/dataset.jsonl -------------------------------------------------------------------------------- /multiencoder/test/test_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/multiencoder/test/test_dataset.py -------------------------------------------------------------------------------- /multiencoder/test/test_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/multiencoder/test/test_model.py -------------------------------------------------------------------------------- /multiencoder/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/multiencoder/train.py -------------------------------------------------------------------------------- /multiencoder/transformers/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/multiencoder/transformers/LICENSE -------------------------------------------------------------------------------- /multiencoder/transformers/changes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/multiencoder/transformers/changes -------------------------------------------------------------------------------- /preprocessing/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/preprocessing/README.md -------------------------------------------------------------------------------- /preprocessing/prep_qmsum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/preprocessing/prep_qmsum.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/requirements.txt -------------------------------------------------------------------------------- /rouge/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /rouge/report_rouge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/query-focused-sum/HEAD/rouge/report_rouge.py --------------------------------------------------------------------------------