TACO: Topics in Algorithmic COde generation dataset

22 Dec 2023  ·  Rongao Li, Jie Fu, Bo-Wen Zhang, Tao Huang, Zhihong Sun, Chen Lyu, Guang Liu, Zhi Jin, Ge Li ·

We introduce TACO, an open-source, large-scale code generation dataset, with a focus on the optics of algorithms, designed to provide a more challenging training dataset and evaluation benchmark in the field of code generation models. TACO includes competition-level programming questions that are more challenging, to enhance or evaluate problem understanding and reasoning abilities in real-world programming scenarios. There are 25433 and 1000 coding problems in training and test set, as well as up to 1.55 million diverse solution answers. Moreover, each TACO problem includes several fine-grained labels such as task topics, algorithms, programming skills, and difficulty levels, providing a more precise reference for the training and evaluation of code generation models. The dataset and evaluation scripts are available on Hugging Face Hub (https://huggingface.co/datasets/BAAI/TACO) and Github (https://github.com/FlagOpen/TACO).

PDF Abstract

Datasets


Introduced in the Paper:

TACO-Code

Used in the Paper:

APPS CodeContests

Results from the Paper


Task Dataset Model Metric Name Metric Value Global Rank Benchmark
Code Generation TACO-Code GPT-4 easy pass@1 31.50% # 1
Code Generation TACO-Code Starcoder-15.5B easy pass@1 11.6% # 2
Code Generation TACO-Code CodeLlama-7B-Python easy pass@1 9.32% # 3

Methods