遇见数据集

Scalable Code Generation with Large Language Models: An Open-Source Ensemble and Reinforcement Learning–Based Selector

收藏
Zenodo2026-06-17 更新2026-05-29 收录
官方服务:

资源简介:

Replication Package: (Mohammed A. Shehab*, Safwan Omari, Mohammad Wardat, and Yaser Jararweh – Accepted) This repository provides the replication materials for the study “Scalable Code Generation with Large Language Models: An Open-Source Ensemble and Reinforcement Learning–Based Selector”. It includes datasets, experiment scripts, and evaluation pipelines used to compare commercial large language models (Codex, Copilot, PaLM2) with two novel open-source methods: Open-Source Ensemble — synthesizes multiple LLM outputs to improve correctness. RL-Based Model Selector — applies reinforcement learning to dynamically select efficient model subsets. 📊 Contents dataset/ – programming tasks, evaluation logs, model outputs, and metrics. ensemble_LLMs.py – ensemble synthesis pipeline. train_rl_weights.py / RL_route_torch.py – reinforcement learning agent training and routing. offline_eval_torch.py / eval_resynthesis_torch.py – offline evaluation utilities. Dataset Information.xlsx – benchmark metadata and summary. ⚙️ Environment Tested with Python 3.9+, PyTorch, Transformers, Gymnasium, Lizard, NumPy, Pandas, and Matplotlib.The requirements.txt file is included for reproducibility. 🔬 Purpose The package supports transparency and independent replication of our experiments on accuracy, code quality, runtime performance, and efficiency (token cost + latency). ⚖️ License Released under the MIT License. All datasets are anonymized and contain no personal or proprietary information.

提供机构:
Zenodo
创建时间:
2025-10-14
二维码
社区交流群
二维码
科研交流群
商业服务