five

LocalDoc/azerbaijani_retriever_corpus-reranked

收藏
Hugging Face2026-03-08 更新2026-03-29 收录
下载链接:
https://hf-mirror.com/datasets/LocalDoc/azerbaijani_retriever_corpus-reranked
下载链接
链接失效反馈
官方服务:
资源简介:
--- language: - az license: cc-by-4.0 tags: - retrieval - reranking - azerbaijani - legislation pretty_name: Azerbaijan Legislation Retrieval Corpus (Reranked) dataset_info: - config_name: corpus features: - name: chunk_id dtype: string - name: passage dtype: string splits: - name: train num_bytes: 67138014 num_examples: 65188 download_size: 37981328 dataset_size: 67138014 - config_name: hard_negatives features: - name: query_id dtype: string - name: chunk_id dtype: string - name: pos_score dtype: float64 - name: neg_1_id dtype: string - name: neg_1_score dtype: float64 - name: neg_2_id dtype: string - name: neg_2_score dtype: float64 - name: neg_3_id dtype: string - name: neg_3_score dtype: float64 - name: neg_4_id dtype: string - name: neg_4_score dtype: float64 - name: neg_5_id dtype: string - name: neg_5_score dtype: float64 - name: neg_6_id dtype: string - name: neg_6_score dtype: float64 - name: neg_7_id dtype: string - name: neg_7_score dtype: float64 - name: neg_8_id dtype: string - name: neg_8_score dtype: float64 - name: neg_9_id dtype: string - name: neg_9_score dtype: float64 - name: neg_10_id dtype: string - name: neg_10_score dtype: float64 splits: - name: train num_bytes: 63959900 num_examples: 188941 download_size: 34604048 dataset_size: 63959900 - config_name: queries features: - name: query_id dtype: string - name: chunk_id dtype: string - name: query dtype: string splits: - name: train num_bytes: 20180731 num_examples: 188941 download_size: 9262163 dataset_size: 20180731 task_categories: - sentence-similarity size_categories: - 10K<n<100K configs: - config_name: corpus data_files: - split: train path: corpus/train-* - config_name: hard_negatives data_files: - split: train path: hard_negatives/train-* - config_name: queries data_files: - split: train path: queries/train-* --- # Azerbaijan Legislation Retrieval Corpus — Reranked Reranked version of [LocalDoc/azerbaijani_retriever_corpus](https://huggingface.co/datasets/LocalDoc/azerbaijani_retriever_corpus). Hard negatives were re-scored with **BAAI/bge-reranker-v2-m3** cross-encoder. False negatives (score > 95% of positive score) were filtered out. Remaining negatives are sorted by score descending (hardest first). ## Configs | Config | Rows | Description | |---|---|---| | `corpus` | 65,188 | Passage chunks: `chunk_id`, `passage` | | `queries` | 188,941 | Queries: `query_id`, `chunk_id`, `query` | | `hard_negatives` | 188,941 | Reranked negatives: `query_id`, `chunk_id`, `pos_score`, `neg_{1..10}_id`, `neg_{1..10}_score` | `query_id` links `queries` and `hard_negatives`. `chunk_id` links to `corpus` (positive passage and negative IDs). ## Usage ```python from datasets import load_dataset corpus = load_dataset("LocalDoc/azerbaijani_retriever_corpus-reranked", "corpus")["train"] queries = load_dataset("LocalDoc/azerbaijani_retriever_corpus-reranked", "queries")["train"] hard_negs = load_dataset("LocalDoc/azerbaijani_retriever_corpus-reranked", "hard_negatives")["train"] # Positive passage for a query q = queries[0] chunk2passage = {r["chunk_id"]: r["passage"] for r in corpus} print(q["query"]) print(chunk2passage[q["chunk_id"]]) # Hard negatives hn = hard_negs[0] for k in range(1, 4): nid = hn[f"neg_{k}_id"] print(f"neg_{k} (score={hn[f'neg_{k}_score']:.4f}): {chunk2passage[nid][:100]}") ``` ## Reranking details - **Model**: `BAAI/bge-reranker-v2-m3` - **Source negatives**: 100 per query (BM25 mined from original dataset) - **False negative filter**: negatives with score > 95% of positive score removed - **Output**: top 10 hardest negatives per query, sorted by descending score

language: - 阿塞拜疆语 license: CC BY 4.0(知识共享署名4.0国际许可协议) tags: - 检索 - 重排序 - 阿塞拜疆语 - 立法 pretty_name: 阿塞拜疆立法检索语料库(重排序版) dataset_info: - config_name: corpus features: - name: chunk_id dtype: 字符串 - name: passage dtype: 字符串 splits: - name: train num_bytes: 67138014 num_examples: 65188 download_size: 37981328 dataset_size: 67138014 - config_name: hard_negatives features: - name: query_id dtype: 字符串 - name: chunk_id dtype: 字符串 - name: pos_score dtype: float64 - name: neg_1_id dtype: 字符串 - name: neg_1_score dtype: float64 - name: neg_2_id dtype: 字符串 - name: neg_2_score dtype: float64 - name: neg_3_id dtype: 字符串 - name: neg_3_score dtype: float64 - name: neg_4_id dtype: 字符串 - name: neg_4_score dtype: float64 - name: neg_5_id dtype: 字符串 - name: neg_5_score dtype: float64 - name: neg_6_id dtype: 字符串 - name: neg_6_score dtype: float64 - name: neg_7_id dtype: 字符串 - name: neg_7_score dtype: float64 - name: neg_8_id dtype: 字符串 - name: neg_8_score dtype: float64 - name: neg_9_id dtype: 字符串 - name: neg_9_score dtype: float64 - name: neg_10_id dtype: 字符串 - name: neg_10_score dtype: float64 splits: - name: train num_bytes: 63959900 num_examples: 188941 download_size: 34604048 dataset_size: 63959900 - config_name: queries features: - name: query_id dtype: 字符串 - name: chunk_id dtype: 字符串 - name: query dtype: 字符串 splits: - name: train num_bytes: 20180731 num_examples: 188941 download_size: 9262163 dataset_size: 20180731 task_categories: - 句子相似度 size_categories: - 10K<n<100K configs: - config_name: corpus data_files: - split: train path: corpus/train-* - config_name: hard_negatives data_files: - split: train path: hard_negatives/train-* - config_name: queries data_files: - split: train path: queries/train-* # 阿塞拜疆立法检索语料库——重排序版 本数据集为[LocalDoc/azerbaijani_retriever_corpus](https://huggingface.co/datasets/LocalDoc/azerbaijani_retriever_corpus)的重排序版本。 负样本集经**BAAI/bge-reranker-v2-m3**交叉编码器(cross-encoder)重新打分。过滤掉得分高于正样本得分95%的假负样本,剩余负样本按得分降序排列(难度由高到低)。 ## 配置项 | 配置名称 | 样本数 | 描述 | |---|---|---| | `corpus` | 65,188 | 段落块:包含`chunk_id`(块ID)、`passage`(段落文本) | | `queries` | 188,941 | 查询样本:包含`query_id`(查询ID)、`chunk_id`(块ID)、`query`(查询文本) | | `hard_negatives` | 188,941 | 重排序负样本:包含`query_id`(查询ID)、`chunk_id`(块ID)、`pos_score`(正样本得分)、`neg_{1..10}_id`(负样本1至10的ID)、`neg_{1..10}_score`(负样本1至10的得分) | `query_id`可关联`queries`与`hard_negatives`配置,`chunk_id`可关联至`corpus`配置(对应正样本段落与负样本ID)。 ## 使用示例 python from datasets import load_dataset corpus = load_dataset("LocalDoc/azerbaijani_retriever_corpus-reranked", "corpus")["train"] queries = load_dataset("LocalDoc/azerbaijani_retriever_corpus-reranked", "queries")["train"] hard_negs = load_dataset("LocalDoc/azerbaijani_retriever_corpus-reranked", "hard_negatives")["train"] # 查询对应的正样本段落 q = queries[0] chunk2passage = {r["chunk_id"]: r["passage"] for r in corpus} print(q["query"]) print(chunk2passage[q["chunk_id"]]) # 难负样本 hn = hard_negs[0] for k in range(1, 4): nid = hn[f"neg_{k}_id"] print(f"neg_{k} (score={hn[f'neg_{k}_score']:.4f}): {chunk2passage[nid][:100]}") ## 重排序细节 - **模型**:`BAAI/bge-reranker-v2-m3` - **原始负样本来源**:每个查询对应100个负样本(通过BM25从原始数据集挖掘得到) - **假负样本过滤规则**:移除得分高于正样本得分95%的负样本 - **输出结果**:每个查询选取得分降序排列的前10个最难负样本
提供机构:
LocalDoc
二维码
社区交流群
二维码
科研交流群
商业服务