Spaces:
Runtime error
Runtime error
We can load HumanEval dataset and pass@k metric from 🤗 [`datasets`](https://huggingface.co/docs/datasets/index) and 🤗 [`evaluate`](https://huggingface.co/docs/evaluate/index) | |
```python | |
from datasets import load_dataset | |
from evaluate import load | |
human_eval = load_dataset("openai_humaneval") | |
code_eval_metric = load("code_eval") | |
``` | |
We can easily compute the pass@k for a problem that asks for the implementation of a function that sums two integers: | |
```python | |
test_cases = ["assert add(2,3)==5"] | |
candidates = [["def add(a,b): return a*b", "def add(a, b): return a+b"]] | |
pass_at_k, results = code_eval_metric.compute(references=test_cases, predictions=candidates, k=[1, 2]) | |
print(pass_at_k) | |
{'pass@1': 0.5, 'pass@2': 1.0} | |
``` | |
To better understand how pass@k metric works, we will illustrate it with a concrete example from HumanEval dataset. We select the problem below and see how CodeParrot 🦜 (110M) performs and which code completions pass the unit tests: | |
**Problem:** | |
```python | |
def truncate_number(number: float) -> float: | |
""" Given a positive floating point number, it can be decomposed into | |
and integer part (largest integer smaller than given number) and decimals | |
(leftover part always smaller than 1). | |
Return the decimal part of the number. | |
>>> truncate_number(3.5) | |
0.5 | |
""" | |
```` | |
Instead of 200 candidate solutions, we will only generate 20 samples for illustration purposes. We use nucleus sampling with top-p where `p=0.95`, `temperature=0.2`, and sample tokens from the model until we encounter a stop sequence indicating the end of a method: ‘\nclass’, ‘\ndef’, ‘\n#’, ‘\nif’, or ‘\nprint’. For more details about decoding strategies for language generation, we recommend this [blog](https://huggingface.co/blog/how-to-generate). | |
**Remark**: | |
Regarding the temperature parameter, in [CodeGen](https://github.com/salesforce/CodeGen) paper, the authors observed that the best performing temperature increases as the number of samples permitted k increases. When a model is only allowed a few samples to pass unit tests, it is beneficial to use the learned distribution, through a low temperature, to select candidates that are likely to pass. But when a model is allowed for more chances with a high k, using a higher sampling temperature to tilt the learned model distribution lets it explore diverse samples and thus have a greater chance of synthesizing a correct program. | |
For our experiment, we compute pass@1, pass@10 and pass@20, each corresponding to unit test pass rate when selecting respectively 1, 10 and 20 samples from the candidate solutions. | |
``` | |
Results: {'pass@1': 0.1, 'pass@10': 0.7631, 'pass@20': 1.0} | |
```` | |
If we take a closer look at the unit test results for each candidate solution, we find that 2 passed the unit test. This means that we have 2 correct solutions among 20, which corresponds to our pass@1 value `2/20 = 0.1`. The scores pass@10 and pass@20 are higher, because the more samples we select from the candidate completions, the more likely we are to include the correct implementation. As | |
for pass@20, it is `1`, since if we select all 20 candidates the problem gets solved which gives 100% success rate. |