arjunguha/mbpp

This is the MBPP dataset. Downloaded from here and constructed as follows: import datasets ds = datasets.load_dataset("json", data_files="mbpp.jsonl", split="train") test = ds.filter(lambda item: item['task_id'] >= 11 and item['task_id'] <= 510) few_shot = ds.filter(lambda item: item['task_id'] >= 1 and item['task_id'] <= 10) validation = ds.filter(lambda item: item['task_id'] >= 511 and item['task_id'] <= 600) train = ds.filter(lambda item: item['task_id'] >= 601 and item['task_id'] <= 974)… See the full description on the dataset page: https://huggingface.co/datasets/arjunguha/mbpp.