Data Science Problems

Introduced by Chandel et al. in Training and Evaluating a Jupyter Notebook Data Science Assistant

Evaluate a natural language code generation model on real data science pedagogical notebooks! Data Science Problems (DSP) includes well-posed data science problems in Markdown along with unit tests to verify correctness and a Docker environment for reproducible execution. About 1/3 of notebooks in this benchmark also include data dependencies, so this benchmark not only can test a model's ability to chain together complex tasks, but also evaluate the solutions on real data! See our paper Training and Evaluating a Jupyter Notebook Data Science Assistant for more details about state of the art results and other properties of the dataset.

Papers


Paper Code Results Date Stars

Tasks


Similar Datasets


License


Modalities


Languages