Sotopia-artifacts
收藏资源简介:
该数据集是一个用于多代理模拟与交互研究的结构化数据集,包含六个独立配置,分别描述代理档案、环境档案、代理关系、环境-代理组合、合成环境档案以及交互事件片段。agent_profiles 配置包含40个样本,详细定义了每个代理的个人属性,包括姓名、年龄、职业、性别、人格特质(如大五人格、MBTI类型)、价值观(道德价值观和施瓦茨个人价值观)、决策风格、秘密及关联模型ID,用于构建具有丰富心理特征的代理。environment_profiles 配置包含90个样本,描述了环境场景,包括代号、来源、场景描述、代理目标、关系强度、年龄与职业约束等,而 environment_profiles_synthetic 是其扩展版本,包含360个样本并增加了任务类型字段。relationship_profiles 配置包含120个样本,定义了代理对之间的双向关系,包括关系类型和背景故事。env_agent_combos 配置包含450个样本,关联环境ID与代理ID列表,用于环境与代理的匹配。episodes 配置是核心交互数据,包含21,333个样本,记录了代理在特定环境中的交互事件,包括消息、推理过程、奖励信号及奖励提示,适用于训练或评估多代理对话、决策模拟等任务。数据集整体支持基于人格的AI代理开发、社交行为分析和交互式系统研究。
This dataset is a structured dataset for multi-agent simulation and interaction research, consisting of six independent configurations: agent profiles, environment profiles, agent relationships, environment-agent combinations, synthetic environment profiles, and interaction event episodes. The agent_profiles configuration contains 40 samples, detailing each agents personal attributes, including name, age, occupation, gender, personality traits (such as Big Five and MBTI types), values (moral values and Schwartz personal values), decision-making styles, secrets, and associated model IDs, used to construct agents with rich psychological characteristics. The environment_profiles configuration contains 90 samples, describing environmental scenarios, including code names, sources, scene descriptions, agent goals, relationship strength, age and occupation constraints, etc., while environment_profiles_synthetic is an extended version with 360 samples and an added task type field. The relationship_profiles configuration contains 120 samples, defining bidirectional relationships between agent pairs, including relationship types and background stories. The env_agent_combos configuration contains 450 samples, linking environment IDs with agent ID lists for environment-agent matching. The episodes configuration is the core interaction data, containing 21,333 samples, recording interaction events of agents in specific environments, including messages, reasoning processes, reward signals, and reward prompts, suitable for training or evaluating multi-agent dialogue, decision simulation, and other tasks. The dataset overall supports personality-based AI agent development, social behavior analysis, and interactive system research.
数据集概述:Sotopia-artifacts
该数据集由 DJ-Research 提供,用于支持社会交互智能体的研究与评估。数据集包含多个配置(config),每个配置对应一类特定的数据。
数据集配置(Configs)
数据集包含以下六个配置,每个配置仅有 train 分割(split),无验证或测试集:
-
agent_profiles(智能体档案)
- 样本数:40
- 特征:包含智能体的基本属性和心理特征,如
pk(主键)、first_name(名)、last_name(姓)、age(年龄)、occupation(职业)、gender(性别)、gender_pronoun(性别代词)、public_info(公开信息)、big_five(大五人格)、moral_values(道德价值观)、schwartz_personal_values(施瓦茨个人价值观)、personality_and_values(性格与价值观)、decision_making_style(决策风格)、secret(秘密)、model_id(模型ID)和mbti(MBTI人格类型)。
-
env_agent_combos(环境与智能体组合)
- 样本数:450
- 特征:定义了特定环境(
env_id)与多个智能体(agent_ids)的组合关系,每条记录通过pk唯一标识。
-
environment_profiles(环境档案)
- 样本数:90
- 特征:描述具体的社会环境,包括
pk、codename(代号)、source(来源)、scenario(场景)、agent_goals(智能体目标列表)、relationship(关系)、age_constraint(年龄约束)、occupation_constraint(职业约束)、agent_constraint(智能体约束)和task_type(任务类型)。
-
environment_profiles_synthetic(合成环境档案)
- 样本数:360
- 特征:与
environment_profiles结构类似,但为合成生成的数据,其中pk和agent_constraint字段为 null。包含codename、source、scenario、agent_goals、relationship、age_constraint、occupation_constraint和task_type。
-
episodes(交互回合)
- 样本数:21333
- 特征:记录智能体在特定环境中的完整交互过程,包含
pk、environment(环境)、agents(智能体)、tag(标签)、models(使用的模型)、messages(消息序列)、reasoning(推理过程)和rewards(奖励值)、rewards_prompt(奖励提示)。
-
relationship_profiles(关系档案)
- 样本数:120
- 特征:定义两个智能体(
agent_1_id与agent_2_id)之间的关系,包括pk、relationship(关系类型)和background_story(背景故事)。
数据规模
| 配置 | 样本数 | 数据集大小 |
|---|---|---|
| agent_profiles | 40 | 23.4 KB |
| env_agent_combos | 450 | 55.8 KB |
| environment_profiles | 90 | 82.6 KB |
| environment_profiles_synthetic | 360 | 250.6 KB |
| episodes | 21333 | 461.3 MB |
| relationship_profiles | 120 | 55.7 KB |
数据集总下载大小约为 109.6 MB,总数据集大小约为 461.8 MB。





