├── .gitignore ├── CITATION.cff ├── LICENSE ├── README.md ├── dataset_helper ├── conala │ ├── execution_eval.py │ └── gen_metric.py └── tldr │ └── gen_metric.py ├── generator └── fid │ ├── LICENSE │ ├── README.md │ ├── check_reload_model.py │ ├── fid_to_reload.py │ ├── requirements.txt │ ├── setup.py │ ├── src │ ├── data.py │ ├── evaluation.py │ ├── index.py │ ├── model.py │ ├── options.py │ ├── preprocess.py │ ├── slurm.py │ └── util.py │ ├── test_reader_simple.py │ ├── train_reader.py │ └── utils │ ├── convert_data.py │ └── save_codet5.py ├── media └── overview.png ├── models ├── generator │ └── .placeholder └── retriever │ └── .placeholder ├── prompts ├── conala_baseline.txt ├── conala_docprompting_oracle_docs.txt ├── conala_docpropmting_retrieved_docs.txt ├── tldr_baseline.txt ├── tldr_docprompting_oracle_docs.txt └── tldr_docprompting_retrieved_docs.txt ├── requirements.txt ├── retriever ├── bm25 │ ├── indexer.py │ └── main.py ├── eval.py └── simcse │ ├── arguments.py │ ├── data_utils.py │ ├── model.py │ ├── run_inference.py │ ├── run_train.py │ ├── trainers.py │ └── utils.py ├── scripts └── tldr_gpt_neo.py └── utils ├── constants.py └── util.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/.gitignore -------------------------------------------------------------------------------- /CITATION.cff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/CITATION.cff -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/README.md -------------------------------------------------------------------------------- /dataset_helper/conala/execution_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/dataset_helper/conala/execution_eval.py -------------------------------------------------------------------------------- /dataset_helper/conala/gen_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/dataset_helper/conala/gen_metric.py -------------------------------------------------------------------------------- /dataset_helper/tldr/gen_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/dataset_helper/tldr/gen_metric.py -------------------------------------------------------------------------------- /generator/fid/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/generator/fid/LICENSE -------------------------------------------------------------------------------- /generator/fid/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/generator/fid/README.md -------------------------------------------------------------------------------- /generator/fid/check_reload_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/generator/fid/check_reload_model.py -------------------------------------------------------------------------------- /generator/fid/fid_to_reload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/generator/fid/fid_to_reload.py -------------------------------------------------------------------------------- /generator/fid/requirements.txt: -------------------------------------------------------------------------------- 1 | numpy 2 | torch 3 | faiss-cpu 4 | transformers==3.0.2 5 | tensorboard 6 | -------------------------------------------------------------------------------- /generator/fid/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/generator/fid/setup.py -------------------------------------------------------------------------------- /generator/fid/src/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/generator/fid/src/data.py -------------------------------------------------------------------------------- /generator/fid/src/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/generator/fid/src/evaluation.py -------------------------------------------------------------------------------- /generator/fid/src/index.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/generator/fid/src/index.py -------------------------------------------------------------------------------- /generator/fid/src/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/generator/fid/src/model.py -------------------------------------------------------------------------------- /generator/fid/src/options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/generator/fid/src/options.py -------------------------------------------------------------------------------- /generator/fid/src/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/generator/fid/src/preprocess.py -------------------------------------------------------------------------------- /generator/fid/src/slurm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/generator/fid/src/slurm.py -------------------------------------------------------------------------------- /generator/fid/src/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/generator/fid/src/util.py -------------------------------------------------------------------------------- /generator/fid/test_reader_simple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/generator/fid/test_reader_simple.py -------------------------------------------------------------------------------- /generator/fid/train_reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/generator/fid/train_reader.py -------------------------------------------------------------------------------- /generator/fid/utils/convert_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/generator/fid/utils/convert_data.py -------------------------------------------------------------------------------- /generator/fid/utils/save_codet5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/generator/fid/utils/save_codet5.py -------------------------------------------------------------------------------- /media/overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/media/overview.png -------------------------------------------------------------------------------- /models/generator/.placeholder: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/retriever/.placeholder: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /prompts/conala_baseline.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/prompts/conala_baseline.txt -------------------------------------------------------------------------------- /prompts/conala_docprompting_oracle_docs.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/prompts/conala_docprompting_oracle_docs.txt -------------------------------------------------------------------------------- /prompts/conala_docpropmting_retrieved_docs.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/prompts/conala_docpropmting_retrieved_docs.txt -------------------------------------------------------------------------------- /prompts/tldr_baseline.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/prompts/tldr_baseline.txt -------------------------------------------------------------------------------- /prompts/tldr_docprompting_oracle_docs.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/prompts/tldr_docprompting_oracle_docs.txt -------------------------------------------------------------------------------- /prompts/tldr_docprompting_retrieved_docs.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/prompts/tldr_docprompting_retrieved_docs.txt -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/requirements.txt -------------------------------------------------------------------------------- /retriever/bm25/indexer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/retriever/bm25/indexer.py -------------------------------------------------------------------------------- /retriever/bm25/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/retriever/bm25/main.py -------------------------------------------------------------------------------- /retriever/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/retriever/eval.py -------------------------------------------------------------------------------- /retriever/simcse/arguments.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/retriever/simcse/arguments.py -------------------------------------------------------------------------------- /retriever/simcse/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/retriever/simcse/data_utils.py -------------------------------------------------------------------------------- /retriever/simcse/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/retriever/simcse/model.py -------------------------------------------------------------------------------- /retriever/simcse/run_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/retriever/simcse/run_inference.py -------------------------------------------------------------------------------- /retriever/simcse/run_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/retriever/simcse/run_train.py -------------------------------------------------------------------------------- /retriever/simcse/trainers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/retriever/simcse/trainers.py -------------------------------------------------------------------------------- /retriever/simcse/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/retriever/simcse/utils.py -------------------------------------------------------------------------------- /scripts/tldr_gpt_neo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/scripts/tldr_gpt_neo.py -------------------------------------------------------------------------------- /utils/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/utils/constants.py -------------------------------------------------------------------------------- /utils/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuyanzhou/docprompting/HEAD/utils/util.py --------------------------------------------------------------------------------