Dataset Descriptions

This dataset contains 34999 examples and is synthetically created to enhance models' coding capabilities.
To generate diverse python coding questions, we expand the methodology in Ge et al., 2024 by using personas to ground the code completion question in real-world scenarios. More details and exact prompts used to construct the dataset can be found in our paper.

• Curated by: Allen Institute for AI
• Paper: TBD
• Repository: TBD
• Language(s) (NLP): English
• License: ODC-BY
• Point of Contact: Faeze Brahman

Loading

from datasets import load_dataset

dataset = load_dataset("allenai/tulu-3-personas-math")["train"]

Dataset Structure

Each example in the dataset contains the standard instruction-tuning data points as follow:

• id (str): a unique identifier
• prompt (str): python programming questions grounded in a given persona/scenario
• messages (list): message format used for supervised fine-tuning (this contains user prompt and assistant response)