CoIR-Retrieval/codefeedback-st
收藏Hugging Face2024-09-12 更新2025-04-08 收录
下载链接:
https://hf-mirror.com/datasets/CoIR-Retrieval/codefeedback-st
下载链接
链接失效反馈官方服务:
资源简介:
该数据集包含三个配置:语料库(corpus)、默认(default)和查询(queries)。每个配置具有特定的特征,如_id、partition、text、title、language和meta_information。语料库和查询配置具有相似的特征,而default配置包括query-id、corpus-id和score。数据集分为不同的部分,每个部分都有字节数和示例数的详细信息。corpus部分为246,229,656字节,包含156,526个示例,default的train部分为3,578,836字节,包含125,220个示例,default的test部分为894,734字节,包含31,306个示例,queries部分为118,682,563字节,包含156,526个示例。每个配置还提到了数据集大小和下载大小。README还包含了如何使用MTEB评估框架对特定任务进行模型评估的说明以及提供的代码片段。
The dataset consists of three configurations: corpus, default, and queries. Each configuration has specific features such as _id, partition, text, title, language, and meta_information. The corpus and queries configurations share similar features, while the default configuration includes query-id, corpus-id, and score. The dataset is split into different parts with detailed information on the number of bytes and examples for each split. The corpus split is 246,229,656 bytes with 156,526 examples, the train split for default is 3,578,836 bytes with 125,220 examples, the test split for default is 894,734 bytes with 31,306 examples, and the queries split is 118,682,563 bytes with 156,526 examples. The dataset sizes and download sizes are also mentioned for each configuration. The README also includes instructions on how to use the MTEB evaluation framework to assess a model on specific tasks using the provided code snippet.
提供机构:
CoIR-Retrieval搜集汇总
数据集介绍

构建方式
CoIR-Retrieval/codefeedback-st数据集源自代码检索领域的权威基准MTEB评估框架,旨在服务于代码与自然语言之间的语义匹配任务。其构建方式基于大规模代码语料库,通过精细化的数据划分形成三个核心配置:corpus、queries和default。corpus配置包含逾15万条代码片段,每条数据均配备唯一标识符、分区信息、文本内容、标题、语言属性及元信息,结构严谨。queries配置对应同等数量的查询语句,与corpus形成配对。default配置则存储了查询与语料之间的关联评分(score),并划分为训练集与测试集,分别包含约12.5万和3.1万条样本,为模型训练与评估提供了标准化的监督信号。
特点
该数据集具备鲜明的结构化与多维度特性。首先,语料与查询均采用统一的字段设计,涵盖文本、标题、语言等关键属性,便于跨语言与跨领域的检索实验。其次,corpus与queries的条目数量完全对等(各156,526条),确保了查询-语料映射的完整性。此外,元信息字段中嵌入资源来源(resource),为数据溯源与质量把控提供了依据。default配置中的评分字段(score)为整数类型,直接反映了查询与代码片段的相关性强度,适用于排序学习与检索精度评测。整体数据规模适中,兼顾了训练效率与模型泛化能力的验证需求。
使用方法
使用该数据集时,建议基于MTEB框架进行加载与评估。用户可通过mteb库调用get_tasks函数,指定任务名称为'CodeFeedbackST',即可自动获取对应的训练与测试数据。推荐搭配Sentence Transformers等编码模型,如'intfloat/e5-base-v2',通过MTEB的run方法执行全流程评估。数据加载支持分片读取,corpus、queries及default配置均以parquet格式存储,便于高效处理。实际应用中,可将查询文本输入模型,生成嵌入向量后与语料库进行相似度比对,利用测试集的评分指标衡量检索性能。该流程简洁规范,适合复现与扩展代码检索领域的实验结果。
背景与挑战
背景概述
在代码智能与信息检索交叉领域,代码检索任务日益成为提升开发者效率与软件工程质量的核心驱动力。CoIR-Retrieval/codefeedback-st数据集由国际研究团队于近年构建,旨在系统性地评估模型对代码片段的语义理解与匹配能力。该数据集聚焦于代码反馈场景,涵盖多种编程语言与自然语言查询,其核心研究问题在于如何通过高质量的检索反馈机制,辅助开发者快速定位相关代码资源。凭借其规模化的标注语料与标准化的MTEB评估框架,该数据集已成为代码检索研究的重要基准,推动了如Sentence-BERT等嵌入模型在该领域的性能优化与评估体系完善。
当前挑战
当前该数据集面临的核心挑战体现在三个层面:首先,代码检索领域长期存在的语义鸿沟问题依然突出,即自然语言查询与代码逻辑之间的映射关系复杂多变,现有模型难以精准捕捉细粒度的功能语义。其次,在数据构建过程中,如何确保跨语言、跨资源代码库的标注一致性与覆盖度面临巨大困难,人工标注成本高昂且易引入偏见。此外,随着代码生成与补全技术的演进,检索任务对实时性与动态上下文适应性的要求日益严苛,现有静态语料难以模拟真实开发环境中的迭代反馈场景,制约了模型泛化能力的进一步提升。
常用场景
经典使用场景
在代码智能与信息检索的交叉领域中,CoIR-Retrieval/codefeedback-st数据集被广泛用作评估和训练代码检索模型的核心基准。其经典应用场景聚焦于代码与自然语言之间的语义匹配任务,研究者通常利用该数据集中的查询-代码对,结合MTEB评估框架,对诸如Sentence-BERT等嵌入模型进行微调与性能检验。通过衡量模型在代码片段与用户意图描述之间的排序准确性,该数据集为代码搜索系统的优化提供了标准化的实验平台。
实际应用
在实际工程应用中,该数据集驱动的代码检索模型被广泛部署于集成开发环境(IDE)的智能代码补全、开源代码仓库的搜索引擎以及开发者问答社区中。例如,通过基于CoIR-Retrieval/codefeedback-st训练的检索系统,开发者能够以自然语言描述快速定位到实现特定功能的代码片段,极大提升了代码复用与调试效率。此外,该数据集还支撑了低资源语言代码搜索、跨语言代码匹配等实用场景,为构建高效、可扩展的代码知识库提供了数据基础。
衍生相关工作
CoIR-Retrieval/codefeedback-st的发布催生了一系列衍生研究工作,其中最具代表性的是基于对比学习的代码-文本联合嵌入方法(如CodeBERT、GraphCodeBERT的后续改进),以及针对代码检索任务设计的排序模型优化策略。此外,该数据集与MTEB评估框架的结合,促使研究者提出了多个针对代码检索的专用评估指标和基线系统。这些工作不仅深化了对代码语义理解的理论认知,还推动了如CodeSearchNet、StackOverflowQA等关联数据集上的跨任务迁移学习研究,形成了代码智能领域内一个活跃的研究分支。
以上内容由遇见数据集搜集并总结生成



