Adam Khakhar, Stephen Mell, Osbert Bastani
arXiv
@misc{khakhar2023pac,
title={PAC Prediction Sets for Large Language Models of Code},
author={Adam Khakhar and Stephen Mell and Osbert Bastani},
year={2023},
eprint={2302.08703},
archivePrefix={arXiv},
primaryClass={cs.LG}
}
- Create prompt and solution (humaneval_dataset_interface.py)
- Get prediction (codex_interface.py, inference.py)
- Parse results into AST (parse_results.py, ast_helper.py)
- Compute optimization (optimize.py, optimize_greedy.py)
- Evaluate and create plots (create_plots.py)