DSGym: Revolutionizing Data Science with Advanced AI Agents

DSGym introduces a groundbreaking framework for training and evaluating data science agents, moving beyond static benchmarks. This innovative system enables the development of data science Agents that can autonomously analyze data, generate insights, and accelerate discovery. It also fosters a standardized, extensible testbed for future advancements.

• •DSGym provides a standardized framework for training data science Agents.
• •It features a modular architecture designed for easy expansion with new tasks, agent scaffolds, and tools.
• •DSGym includes a training set and 4B model that outperforms GPT-4o on analysis benchmarks.