当前位置:首页 > 文章列表 > 科技周边 > 人工智能 > Hugging Face Datasets 批量重命名列遇到重名怎样处理

Hugging Face Datasets 批量重命名列遇到重名怎样处理

来源:17golang原创 2026-11-10 18:32:21 0浏览 收藏

批量重命名列遇到重名时,先算出整张Dataset的最终列名,确认每个名称只对应一列,再调用rename_columns。两个旧列指向同一新名,以及新名撞上未改名列,都应在业务侧明确解决;重命名不会替你决定合并还是丢弃数据。

官方文档:https://huggingface.co/docs/datasets/en/package_reference/main_classes#datasets.Dataset.rename_columns

不要只看映射里的目标名称

假设原列为source、target、group。把source改为target,同时保留原target,就会产生两个target。即使映射字典只有一个值,也不能说明最终名称唯一。

原始列映射字典保留列与最终列名集合
列名组成说明图:映射列与未改名列共同决定最终列名,非运行截图。

当前官方实现会拒绝不存在的原列、重复的映射目标及空目标名。不同版本和底层路径的错误表现不应作为自动合并契约;应用代码仍应检查整张表的最终名称。

先解决冲突,再一次性改名

下面给出独立的输入处理示例。这些检查属于应用层数据处理逻辑;示例没有在本机执行。

from collections import Counter
from datasets import Dataset

# 演示表只在内存中创建,不读取外部语料
ds = Dataset.from_dict({"source": ["甲"], "target": ["乙"], "group": [1]})
mapping = {"source": "input_text", "target": "output_text"}

# 应用层先计算整张表的最终名称,而不只检查字典的值
unknown = set(mapping) - set(ds.column_names)
if unknown:
    raise ValueError(f"找不到原列:{sorted(unknown)}")
if any(not isinstance(v, str) or not v for v in mapping.values()):
    raise ValueError("新列名必须是非空字符串")
final_names = [mapping.get(name, name) for name in ds.column_names]
duplicates = [name for name, count in Counter(final_names).items() if count > 1]
if duplicates:
    raise ValueError(f"最终列名冲突:{duplicates}")

# 接住返回的新Dataset,原变量ds继续保留
renamed = ds.rename_columns(mapping)
print(renamed.column_names)  # 预期:input_text、output_text、group

若确实不需要已有target,应先确认数据归属,再显式移除那一列;若两列都要保留,给它们不同的新名称;若想合并值,需要另写合并规则。不能通过重复列名表达优先级。

交换列名用一次完整映射

# 明确交换名称,两个最终名称仍然不同
swapped = ds.rename_columns({"source": "target", "target": "source"})
# 原source的数据现在由target名称访问
print(swapped["target"][0])  # 预期:甲

逐次单列改名会先撞上尚未腾出的名称;完整映射表达的是对原列名的统一替换。这里交换的是名称归属,不是将两列值混合。

source与target列名交换对应关系
列名交换关系说明图:旧source对应新target,旧target对应新source;不是执行流程。

改名后更新下游字段约定

rename_columns返回改名后的Dataset,并将相关features移到新名称下。训练脚本、预处理函数和导出配置中使用的字段名应同步更新。建议把映射作为一次数据准备操作的明确输入保存,避免上游字段变化时静默跳过。

两个旧列可以改成同一个新列吗?

不能把它当成合并操作。先定义值的合并和缺失值处理规则,再构造目标列。

DatasetDict可以直接照搬吗?

批量改名会作用于各split。先明确每个split的列是否一致,不能只用train的列名推断其他split。

实现参考:https://github.com/huggingface/datasets/blob/main/src/datasets/arrow_dataset.py。

版本声明
本文转载于:17golang原创 如有侵犯,请联系study_golang@163.com删除
Redis LCS 的 LEN 和 IDX 在返回内容上有什么区别Redis LCS 的 LEN 和 IDX 在返回内容上有什么区别
上一篇
Redis LCS 的 LEN 和 IDX 在返回内容上有什么区别
Go types.Selection.Recv 怎样保留选择表达式的接收者类型
下一篇
Go types.Selection.Recv 怎样保留选择表达式的接收者类型
查看更多
最新文章
查看更多
课程推荐
  • 前端进阶之JavaScript设计模式
    前端进阶之JavaScript设计模式
    设计模式是开发人员在软件开发过程中面临一般问题时的解决方案,代表了最佳的实践。本课程的主打内容包括JS常见设计模式以及具体应用场景,打造一站式知识长龙服务,适合有JS基础的同学学习。
    543次学习
  • GO语言核心编程课程
    GO语言核心编程课程
    本课程采用真实案例,全面具体可落地,从理论到实践,一步一步将GO核心编程技术、编程思想、底层实现融会贯通,使学习者贴近时代脉搏,做IT互联网时代的弄潮儿。
    516次学习
  • 简单聊聊mysql8与网络通信
    简单聊聊mysql8与网络通信
    如有问题加微信:Le-studyg;在课程中,我们将首先介绍MySQL8的新特性,包括性能优化、安全增强、新数据类型等,帮助学生快速熟悉MySQL8的最新功能。接着,我们将深入解析MySQL的网络通信机制,包括协议、连接管理、数据传输等,让
    500次学习
  • JavaScript正则表达式基础与实战
    JavaScript正则表达式基础与实战
    在任何一门编程语言中,正则表达式,都是一项重要的知识,它提供了高效的字符串匹配与捕获机制,可以极大的简化程序设计。
    487次学习
  • 从零制作响应式网站—Grid布局
    从零制作响应式网站—Grid布局
    本系列教程将展示从零制作一个假想的网络科技公司官网,分为导航,轮播,关于我们,成功案例,服务流程,团队介绍,数据部分,公司动态,底部信息等内容区块。网站整体采用CSSGrid布局,支持响应式,有流畅过渡和展现动画。
    485次学习
查看更多
AI推荐
  • PubMedQA数据集详解:生物医学问答基准、功能与应用指南
    PubMedQA
    深入了解PubMedQA生物医学问答数据集,涵盖其核心功能、使用方法及在临床决策、药物研发等场景的应用,助力提升NLP模型性能。
    968次使用
  • H2O EvalGPT:开源LLM大模型评估与排行榜工具
    H2O EvalGPT
    H2O EvalGPT是H2O.ai推出的开源LLM评估平台,提供详细的大模型性能排行榜、行业特定基准测试及A/B测试功能,助您快速选择最适合项目的高性能大语言模型。
    977次使用
  • LMArena是什么?伯克利AI模型评估平台使用指南与功能解析
    LMArena
    LMArena是加州大学伯克利分校推出的AI模型匿名评测平台。通过盲测投票机制,用户可对比不同大模型回答并生成实时排行榜,助力开发者优化模型及用户选择最佳AI工具。
    1085次使用
  • 斯坦福HELM:大语言模型Holistic Evaluation整体评估框架详解
    HELM
    深入了解斯坦福推出的HELM(Holistic Evaluation of Language Models)大模型评测体系。本文解析其核心功能、安装配置步骤及应用场景,涵盖准确性、公平性、鲁棒性等多维度指标,助力开发者全面优化语言模型性能。
    1061次使用
  • MMBench详解:多模态大模型基准测试、功能特点与使用指南
    MMBench
    MMBench是由上海人工智能实验室等机构联合推出的多模态基准测试平台,提供细粒度能力评估、大规模数据集及VLMEvalKit工具。本文详细介绍其核心功能、安装使用方法及应用场景,助力开发者全面评估多模态模型性能。
    899次使用
微信登录更方便
  • 密码登录
  • 注册账号
登录即同意 用户协议 和 隐私政策
返回登录
  • 重置密码