TML-Bench: Benchmark for Data Science Agents on Tabular ML Tasks
Organizations: Independent Researcher, San Jose, USA
Abstract
Autonomous coding agents can produce strong tabular baselines quickly on Kaggle-style tasks. Practical value depends on end-to-end correctness and reliability under time limits. This paper introduces TML-Bench, a tabular benchmark for data science agents on Kaggle-style tasks. This paper evaluates 10 OSS LLMs on four Kaggle competitions and three time budgets (240s, 600s, and 1200s). Each model is run five times per task and budget. A run is successful if it produces a valid submission and a private-holdout score on hidden labels that are not accessible to the agent. This paper reports median performance, success rates, and run-to-run variability. MiniMax-M2.1 model achieves the best aggregate performance score on all four competitions under the paper's primary aggregation. Average performance improves with larger time budgets. Scaling is noisy for some individual models at the current run count. Code and materials are available at https://github.com/MykolaPinchuk/TML-bench/tree/master.
Figures & tables
Appendix figures & tables12 assets
Supplementary material from the paper’s appendix.
Appendix
| Model | Provider | Type | Parameters | Release date | Knowledge cutoff | License |
|---|---|---|---|---|---|---|
| Qwen3-Coder-480B-A35B | chutes | open weights | 480B total, 35B active | 2025-07-23 | 2025-01-22 | Apache-2.0 |
| gpt-oss-120b | chutes | open weights | 120B | 2025-08-05 | 2024-06-01 | Apache-2.0 |
| GLM-4.7-FP8 | chutes | open weights | 358B | 2025-12-22 | 2025-04 | MIT |
| GLM-4.7-Flash | chutes | open weights | 30B | 2025-12-22 | 2025-04 | MIT |
| MiniMax-M2.1-TEE | chutes | open weights | 230B total, 10B active | 2025-12-23 | 2025-06-25 | unknown |
| GLM-4.6-FP8 | chutes | open weights | 357B | 2025-09-30 | 2025-04 | MIT |
| Competition | Problem and target | Metric | Train rows | Test rows | Features |
|---|---|---|---|---|---|
| bank-customer-churn-ict-u-ai | Binary classification: predict customer churn ( Exited ) | AUC | 12,000 | 3,000 | 12 |
| foot-traffic-wuerzburg-retail-forecasting-2-0 | Regression: forecast foot traffic ( target ) | RMSE | 41,155 | 10,289 | 4 |
| playground-series-s5e10 | Regression: predict road accident risk ( accident_risk ) | RMSE | 414,203 | 103,551 | 12 |
| playground-series-s6e1 | Regression: predict exam scores ( exam_score ) | RMSE | 504,000 | 126,000 | 11 |
AI Agent Benchmarks
AI agent benchmarks measure how reliably autonomous systems plan, use tools, interact, and complete tasks across controlled and real-world settings.
See all 1,494 papers