├── .github └── workflows │ └── pr-arena-workflow.yml ├── .gitignore ├── LICENSE ├── README.md ├── __init__.py ├── apidocs ├── doc_parser.py ├── index_es.py ├── processed │ ├── direct │ │ ├── intent5.jsonl │ │ └── snippet5.jsonl │ └── distsmpl │ │ ├── goldmine_intent_count100k_topk1.txt │ │ ├── goldmine_intent_count100k_topk5.txt │ │ ├── goldmine_snippet_count100k_topk1.txt │ │ ├── goldmine_snippet_count100k_topk5.txt │ │ ├── intent_15k │ │ ├── goldmine_intent_count100k_topk1_temp10.jsonl │ │ ├── goldmine_intent_count100k_topk1_temp2.jsonl │ │ ├── goldmine_intent_count100k_topk1_temp5.jsonl │ │ ├── goldmine_intent_count100k_topk5_temp10.jsonl │ │ ├── goldmine_intent_count100k_topk5_temp2.jsonl │ │ └── goldmine_intent_count100k_topk5_temp5.jsonl │ │ └── snippet_15k │ │ ├── goldmine_snippet_count100k_topk1_temp10.jsonl │ │ ├── goldmine_snippet_count100k_topk1_temp2.jsonl │ │ ├── goldmine_snippet_count100k_topk1_temp5.jsonl │ │ ├── goldmine_snippet_count100k_topk5_temp10.jsonl │ │ ├── goldmine_snippet_count100k_topk5_temp2.jsonl │ │ └── goldmine_snippet_count100k_topk5_temp5.jsonl ├── python-docs.jsonl ├── retrieve.py ├── search_es.py ├── stats.txt └── tag_code_type.py ├── asdl ├── README.md ├── __init__.py ├── asdl.py ├── asdl_ast.py ├── hypothesis.py ├── lang │ ├── __init__.py │ ├── lambda_dcs │ │ ├── __init__.py │ │ ├── lambda_asdl.txt │ │ ├── lambda_dcs_transition_system.py │ │ └── logical_form.py │ ├── prolog │ │ ├── __init__.py │ │ ├── prolog_asdl.txt │ │ └── prolog_transition_system.py │ ├── py │ │ ├── __init__.py │ │ ├── example.py │ │ ├── py_asdl.txt │ │ ├── py_asdl_helper.py │ │ ├── py_transition_system.py │ │ └── py_utils.py │ └── py3 │ │ ├── __init__.py │ │ ├── demo.py │ │ ├── py3_asdl.simplified.txt │ │ ├── py3_asdl.txt │ │ └── py3_transition_system.py ├── transition_system.py └── utils.py ├── best_pretrained_models ├── finetune.mined.retapi.distsmpl.dr0.3.lr0.001.lr_de0.5.lr_da15.beam15.seed0.mined_100000.intent_count100k_topk1_temp5.bin ├── reranker.conala.vocab.src_freq3.code_freq3.mined_100000.intent_count100k_topk1_temp5.bin ├── reranker.conala.vocab.src_freq3.code_freq3.mined_100000.intent_count100k_topk1_temp5.bin.paraphrase_identifier └── reranker.conala.vocab.src_freq3.code_freq3.mined_100000.intent_count100k_topk1_temp5.bin.reconstructor ├── common ├── __init__.py ├── registerable.py ├── savable.py └── utils.py ├── components ├── __init__.py ├── action_info.py ├── dataset.py ├── decode_hypothesis.py ├── evaluator.py ├── reranker.py ├── standalone_parser.py └── vocab.py ├── config ├── conda_environment.yml └── config_conala.json ├── datasets ├── __init__.py ├── conala │ ├── __init__.py │ ├── bleu_score.py │ ├── conala_eval.py │ ├── dataset.py │ ├── docstring.py │ ├── evaluator.py │ ├── example_processor.py │ └── util.py └── utils.py ├── doc ├── api.png ├── approach.png ├── result.png └── system.png ├── evaluation.py ├── exp.py ├── model ├── __init__.py ├── attention_util.py ├── decomposable_attention_model.py ├── lstm.py ├── neural_lm.py ├── nn_utils.py ├── paraphrase.py ├── parser.py ├── pointer_net.py ├── prior.py ├── reconstruction_model.py ├── seq2seq.py ├── seq2seq_copy.py ├── struct_vae.py └── utils.py ├── scripts └── conala │ ├── decode.sh │ ├── finetune.sh │ ├── finetune2.sh │ ├── finetune_retrieved.sh │ ├── finetune_retrieved_distsmpl.sh │ ├── finetune_retrieved_only.sh │ ├── pretrain2.sh │ ├── rerank.sh │ ├── test.sh │ ├── train.sh │ ├── train_apidocs.sh │ ├── train_para.sh │ ├── train_recon.sh │ ├── train_retrieved.sh │ ├── train_retrieved_distsmpl.sh │ ├── train_retrieved_only.sh │ └── vanilla.sh └── server ├── README.md ├── __init__.py ├── app.py ├── static ├── d3Tree.js ├── parser.js └── tree-viewer.css └── templates └── default.html /.github/workflows/pr-arena-workflow.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/.github/workflows/pr-arena-workflow.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- 1 | # coding=utf-8 2 | -------------------------------------------------------------------------------- /apidocs/doc_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/apidocs/doc_parser.py -------------------------------------------------------------------------------- /apidocs/index_es.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/apidocs/index_es.py -------------------------------------------------------------------------------- /apidocs/processed/direct/intent5.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/apidocs/processed/direct/intent5.jsonl -------------------------------------------------------------------------------- /apidocs/processed/direct/snippet5.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/apidocs/processed/direct/snippet5.jsonl -------------------------------------------------------------------------------- /apidocs/processed/distsmpl/goldmine_intent_count100k_topk1.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/apidocs/processed/distsmpl/goldmine_intent_count100k_topk1.txt -------------------------------------------------------------------------------- /apidocs/processed/distsmpl/goldmine_intent_count100k_topk5.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/apidocs/processed/distsmpl/goldmine_intent_count100k_topk5.txt -------------------------------------------------------------------------------- /apidocs/processed/distsmpl/goldmine_snippet_count100k_topk1.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/apidocs/processed/distsmpl/goldmine_snippet_count100k_topk1.txt -------------------------------------------------------------------------------- /apidocs/processed/distsmpl/goldmine_snippet_count100k_topk5.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/apidocs/processed/distsmpl/goldmine_snippet_count100k_topk5.txt -------------------------------------------------------------------------------- /apidocs/processed/distsmpl/intent_15k/goldmine_intent_count100k_topk1_temp10.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/apidocs/processed/distsmpl/intent_15k/goldmine_intent_count100k_topk1_temp10.jsonl -------------------------------------------------------------------------------- /apidocs/processed/distsmpl/intent_15k/goldmine_intent_count100k_topk1_temp2.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/apidocs/processed/distsmpl/intent_15k/goldmine_intent_count100k_topk1_temp2.jsonl -------------------------------------------------------------------------------- /apidocs/processed/distsmpl/intent_15k/goldmine_intent_count100k_topk1_temp5.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/apidocs/processed/distsmpl/intent_15k/goldmine_intent_count100k_topk1_temp5.jsonl -------------------------------------------------------------------------------- /apidocs/processed/distsmpl/intent_15k/goldmine_intent_count100k_topk5_temp10.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/apidocs/processed/distsmpl/intent_15k/goldmine_intent_count100k_topk5_temp10.jsonl -------------------------------------------------------------------------------- /apidocs/processed/distsmpl/intent_15k/goldmine_intent_count100k_topk5_temp2.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/apidocs/processed/distsmpl/intent_15k/goldmine_intent_count100k_topk5_temp2.jsonl -------------------------------------------------------------------------------- /apidocs/processed/distsmpl/intent_15k/goldmine_intent_count100k_topk5_temp5.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/apidocs/processed/distsmpl/intent_15k/goldmine_intent_count100k_topk5_temp5.jsonl -------------------------------------------------------------------------------- /apidocs/processed/distsmpl/snippet_15k/goldmine_snippet_count100k_topk1_temp10.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/apidocs/processed/distsmpl/snippet_15k/goldmine_snippet_count100k_topk1_temp10.jsonl -------------------------------------------------------------------------------- /apidocs/processed/distsmpl/snippet_15k/goldmine_snippet_count100k_topk1_temp2.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/apidocs/processed/distsmpl/snippet_15k/goldmine_snippet_count100k_topk1_temp2.jsonl -------------------------------------------------------------------------------- /apidocs/processed/distsmpl/snippet_15k/goldmine_snippet_count100k_topk1_temp5.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/apidocs/processed/distsmpl/snippet_15k/goldmine_snippet_count100k_topk1_temp5.jsonl -------------------------------------------------------------------------------- /apidocs/processed/distsmpl/snippet_15k/goldmine_snippet_count100k_topk5_temp10.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/apidocs/processed/distsmpl/snippet_15k/goldmine_snippet_count100k_topk5_temp10.jsonl -------------------------------------------------------------------------------- /apidocs/processed/distsmpl/snippet_15k/goldmine_snippet_count100k_topk5_temp2.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/apidocs/processed/distsmpl/snippet_15k/goldmine_snippet_count100k_topk5_temp2.jsonl -------------------------------------------------------------------------------- /apidocs/processed/distsmpl/snippet_15k/goldmine_snippet_count100k_topk5_temp5.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/apidocs/processed/distsmpl/snippet_15k/goldmine_snippet_count100k_topk5_temp5.jsonl -------------------------------------------------------------------------------- /apidocs/python-docs.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/apidocs/python-docs.jsonl -------------------------------------------------------------------------------- /apidocs/retrieve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/apidocs/retrieve.py -------------------------------------------------------------------------------- /apidocs/search_es.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/apidocs/search_es.py -------------------------------------------------------------------------------- /apidocs/stats.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/apidocs/stats.txt -------------------------------------------------------------------------------- /apidocs/tag_code_type.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/apidocs/tag_code_type.py -------------------------------------------------------------------------------- /asdl/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/asdl/README.md -------------------------------------------------------------------------------- /asdl/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/asdl/__init__.py -------------------------------------------------------------------------------- /asdl/asdl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/asdl/asdl.py -------------------------------------------------------------------------------- /asdl/asdl_ast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/asdl/asdl_ast.py -------------------------------------------------------------------------------- /asdl/hypothesis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/asdl/hypothesis.py -------------------------------------------------------------------------------- /asdl/lang/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /asdl/lang/lambda_dcs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /asdl/lang/lambda_dcs/lambda_asdl.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/asdl/lang/lambda_dcs/lambda_asdl.txt -------------------------------------------------------------------------------- /asdl/lang/lambda_dcs/lambda_dcs_transition_system.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/asdl/lang/lambda_dcs/lambda_dcs_transition_system.py -------------------------------------------------------------------------------- /asdl/lang/lambda_dcs/logical_form.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/asdl/lang/lambda_dcs/logical_form.py -------------------------------------------------------------------------------- /asdl/lang/prolog/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /asdl/lang/prolog/prolog_asdl.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/asdl/lang/prolog/prolog_asdl.txt -------------------------------------------------------------------------------- /asdl/lang/prolog/prolog_transition_system.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/asdl/lang/prolog/prolog_transition_system.py -------------------------------------------------------------------------------- /asdl/lang/py/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /asdl/lang/py/example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/asdl/lang/py/example.py -------------------------------------------------------------------------------- /asdl/lang/py/py_asdl.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/asdl/lang/py/py_asdl.txt -------------------------------------------------------------------------------- /asdl/lang/py/py_asdl_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/asdl/lang/py/py_asdl_helper.py -------------------------------------------------------------------------------- /asdl/lang/py/py_transition_system.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/asdl/lang/py/py_transition_system.py -------------------------------------------------------------------------------- /asdl/lang/py/py_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/asdl/lang/py/py_utils.py -------------------------------------------------------------------------------- /asdl/lang/py3/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /asdl/lang/py3/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/asdl/lang/py3/demo.py -------------------------------------------------------------------------------- /asdl/lang/py3/py3_asdl.simplified.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/asdl/lang/py3/py3_asdl.simplified.txt -------------------------------------------------------------------------------- /asdl/lang/py3/py3_asdl.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/asdl/lang/py3/py3_asdl.txt -------------------------------------------------------------------------------- /asdl/lang/py3/py3_transition_system.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/asdl/lang/py3/py3_transition_system.py -------------------------------------------------------------------------------- /asdl/transition_system.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/asdl/transition_system.py -------------------------------------------------------------------------------- /asdl/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/asdl/utils.py -------------------------------------------------------------------------------- /best_pretrained_models/finetune.mined.retapi.distsmpl.dr0.3.lr0.001.lr_de0.5.lr_da15.beam15.seed0.mined_100000.intent_count100k_topk1_temp5.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/best_pretrained_models/finetune.mined.retapi.distsmpl.dr0.3.lr0.001.lr_de0.5.lr_da15.beam15.seed0.mined_100000.intent_count100k_topk1_temp5.bin -------------------------------------------------------------------------------- /best_pretrained_models/reranker.conala.vocab.src_freq3.code_freq3.mined_100000.intent_count100k_topk1_temp5.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/best_pretrained_models/reranker.conala.vocab.src_freq3.code_freq3.mined_100000.intent_count100k_topk1_temp5.bin -------------------------------------------------------------------------------- /best_pretrained_models/reranker.conala.vocab.src_freq3.code_freq3.mined_100000.intent_count100k_topk1_temp5.bin.paraphrase_identifier: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/best_pretrained_models/reranker.conala.vocab.src_freq3.code_freq3.mined_100000.intent_count100k_topk1_temp5.bin.paraphrase_identifier -------------------------------------------------------------------------------- /best_pretrained_models/reranker.conala.vocab.src_freq3.code_freq3.mined_100000.intent_count100k_topk1_temp5.bin.reconstructor: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/best_pretrained_models/reranker.conala.vocab.src_freq3.code_freq3.mined_100000.intent_count100k_topk1_temp5.bin.reconstructor -------------------------------------------------------------------------------- /common/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /common/registerable.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/common/registerable.py -------------------------------------------------------------------------------- /common/savable.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/common/savable.py -------------------------------------------------------------------------------- /common/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/common/utils.py -------------------------------------------------------------------------------- /components/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/components/__init__.py -------------------------------------------------------------------------------- /components/action_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/components/action_info.py -------------------------------------------------------------------------------- /components/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/components/dataset.py -------------------------------------------------------------------------------- /components/decode_hypothesis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/components/decode_hypothesis.py -------------------------------------------------------------------------------- /components/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/components/evaluator.py -------------------------------------------------------------------------------- /components/reranker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/components/reranker.py -------------------------------------------------------------------------------- /components/standalone_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/components/standalone_parser.py -------------------------------------------------------------------------------- /components/vocab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/components/vocab.py -------------------------------------------------------------------------------- /config/conda_environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/config/conda_environment.yml -------------------------------------------------------------------------------- /config/config_conala.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/config/config_conala.json -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /datasets/conala/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /datasets/conala/bleu_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/datasets/conala/bleu_score.py -------------------------------------------------------------------------------- /datasets/conala/conala_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/datasets/conala/conala_eval.py -------------------------------------------------------------------------------- /datasets/conala/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/datasets/conala/dataset.py -------------------------------------------------------------------------------- /datasets/conala/docstring.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/datasets/conala/docstring.py -------------------------------------------------------------------------------- /datasets/conala/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/datasets/conala/evaluator.py -------------------------------------------------------------------------------- /datasets/conala/example_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/datasets/conala/example_processor.py -------------------------------------------------------------------------------- /datasets/conala/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/datasets/conala/util.py -------------------------------------------------------------------------------- /datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/datasets/utils.py -------------------------------------------------------------------------------- /doc/api.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/doc/api.png -------------------------------------------------------------------------------- /doc/approach.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/doc/approach.png -------------------------------------------------------------------------------- /doc/result.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/doc/result.png -------------------------------------------------------------------------------- /doc/system.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/doc/system.png -------------------------------------------------------------------------------- /evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/evaluation.py -------------------------------------------------------------------------------- /exp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/exp.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/attention_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/model/attention_util.py -------------------------------------------------------------------------------- /model/decomposable_attention_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/model/decomposable_attention_model.py -------------------------------------------------------------------------------- /model/lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/model/lstm.py -------------------------------------------------------------------------------- /model/neural_lm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/model/neural_lm.py -------------------------------------------------------------------------------- /model/nn_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/model/nn_utils.py -------------------------------------------------------------------------------- /model/paraphrase.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/model/paraphrase.py -------------------------------------------------------------------------------- /model/parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/model/parser.py -------------------------------------------------------------------------------- /model/pointer_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/model/pointer_net.py -------------------------------------------------------------------------------- /model/prior.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/model/prior.py -------------------------------------------------------------------------------- /model/reconstruction_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/model/reconstruction_model.py -------------------------------------------------------------------------------- /model/seq2seq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/model/seq2seq.py -------------------------------------------------------------------------------- /model/seq2seq_copy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/model/seq2seq_copy.py -------------------------------------------------------------------------------- /model/struct_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/model/struct_vae.py -------------------------------------------------------------------------------- /model/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/model/utils.py -------------------------------------------------------------------------------- /scripts/conala/decode.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/scripts/conala/decode.sh -------------------------------------------------------------------------------- /scripts/conala/finetune.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/scripts/conala/finetune.sh -------------------------------------------------------------------------------- /scripts/conala/finetune2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/scripts/conala/finetune2.sh -------------------------------------------------------------------------------- /scripts/conala/finetune_retrieved.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/scripts/conala/finetune_retrieved.sh -------------------------------------------------------------------------------- /scripts/conala/finetune_retrieved_distsmpl.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/scripts/conala/finetune_retrieved_distsmpl.sh -------------------------------------------------------------------------------- /scripts/conala/finetune_retrieved_only.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/scripts/conala/finetune_retrieved_only.sh -------------------------------------------------------------------------------- /scripts/conala/pretrain2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/scripts/conala/pretrain2.sh -------------------------------------------------------------------------------- /scripts/conala/rerank.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/scripts/conala/rerank.sh -------------------------------------------------------------------------------- /scripts/conala/test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/scripts/conala/test.sh -------------------------------------------------------------------------------- /scripts/conala/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/scripts/conala/train.sh -------------------------------------------------------------------------------- /scripts/conala/train_apidocs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/scripts/conala/train_apidocs.sh -------------------------------------------------------------------------------- /scripts/conala/train_para.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/scripts/conala/train_para.sh -------------------------------------------------------------------------------- /scripts/conala/train_recon.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/scripts/conala/train_recon.sh -------------------------------------------------------------------------------- /scripts/conala/train_retrieved.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/scripts/conala/train_retrieved.sh -------------------------------------------------------------------------------- /scripts/conala/train_retrieved_distsmpl.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/scripts/conala/train_retrieved_distsmpl.sh -------------------------------------------------------------------------------- /scripts/conala/train_retrieved_only.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/scripts/conala/train_retrieved_only.sh -------------------------------------------------------------------------------- /scripts/conala/vanilla.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/scripts/conala/vanilla.sh -------------------------------------------------------------------------------- /server/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/server/README.md -------------------------------------------------------------------------------- /server/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /server/app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/server/app.py -------------------------------------------------------------------------------- /server/static/d3Tree.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/server/static/d3Tree.js -------------------------------------------------------------------------------- /server/static/parser.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/server/static/parser.js -------------------------------------------------------------------------------- /server/static/tree-viewer.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/server/static/tree-viewer.css -------------------------------------------------------------------------------- /server/templates/default.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neulab/external-knowledge-codegen/HEAD/server/templates/default.html --------------------------------------------------------------------------------