PTBench
收藏资源简介:
PTBench是第一个用于评估大型模型在个性化工具调用方面的能力的基准数据集。该数据集由1083个高质量的标注数据样本组成,旨在解决在工具调用中考虑用户个性化偏好的问题,包括工具偏好和基于用户资料的查询。数据集的构建过程包括三个主要阶段:工具生成、用户资料构建和查询与解决方案生成。通过使用高级大型语言模型(LLM)进行工具生成和用户行为模拟,PTBench提供了多样化的工具和用户资料,以训练和评估模型的个性化工具调用能力。
PTBench is the first benchmark dataset for evaluating the capability of large models in personalized tool invocation. This dataset consists of 1083 high-quality annotated data samples, aiming to address the problem of considering users' personalized preferences during tool invocation, including tool preferences and user profile-based queries. The construction process of the dataset includes three main stages: tool generation, user profile construction, and query and solution generation. By leveraging state-of-the-art large language models (LLMs) for tool generation and user behavior simulation, PTBench provides diverse tools and user profiles for training and evaluating models' capabilities in personalized tool invocation.
PTBench数据集概述
简介
PTBench是首个面向个性化工具调用的基准测试,旨在全面评估模型基于用户信息调用工具的能力。该数据集通过系统化的个性化数据合成框架构建而成。
主要特性
- 支持开源模型(OSS)和API模型的评估
- 提供结果分析和错误统计功能
- 包含在线排行榜系统
支持的开源模型
| 模型名称 | 模板标识 |
|---|---|
| Qwen2.5 | qwen |
| Llama 3 | llama3 |
| Mistral | mistral |
| xLAM | xlam |
| Hammer | hammer |
| deepseek R1(Distill) | deepseek3 |
评估流程
-
环境配置
- 需要Python 3.9环境
- 通过
requirements.txt安装依赖
-
模型评估
- OSS模型:修改
config.yaml配置后运行python run.py --type oss - API模型:设置API密钥后运行
python run.py --type api
- OSS模型:修改
-
结果查看
- 输出分为三部分:未训练用户、已训练用户和总体结果
- 包含准确率和错误分析数据
扩展支持
- 可添加未支持模型:需在
src/template.py中添加模板 - 可修改答案格式:需同步调整
src/parser.py中的解析逻辑
排行榜更新
用户可通过发送邮件至huang_yuefeng@mail.ustc.edu.cn提交评估结果,要求包含:
- HuggingFace上的公开模型名称
- 评估结果CSV文件




