linxy/ICEWS05_15
收藏数据集概述
基本信息
- 许可证: Apache-2.0
- 任务类别: 图机器学习(graph-ml)
- 语言: 英语(en)
- 数据集大小: 1M<n<10M
数据集内容
- 构建基础: 基于ICEWS和GDELT,这两个数据集在时间知识图谱补全(TKGC)领域广泛使用。
- 首次引入: 在论文"TFLEX: Temporal Feature-Logic Embedding Framework for Complex Reasoning over Temporal Knowledge Graph"中首次介绍。
数据集结构
-
数据加载示例: python
dataset = load_dataset("linxy/ICEWS05_15", "all") len(dataset["train"]) + len(dataset["validation"]) + len(dataset["test"]) 4651939
-
数据集样本示例: python {query_name: Pe, definition: def Pe(e1, r1, t1): return Pe(e1, r1, t1), query: [3751, 125, 1330], answer: [10136], easy_answer: [], args: [e1, r1, t1]}
-
元信息加载示例: python {dataset: ICEWS05_15, entity_count: 10488, relation_count: 251, timestamp_count: 4017, valid_triples_count: 46275, test_triples_count: 46092, train_triples_count: 368962, triple_count: 461329, query_meta: {...}, entity2idx: {...}, relation2idx: {...}, timestamp2idx: {...}}
数据集统计
- 查询计数统计: 提供不同查询类型的训练、验证和测试集的查询数量。
- 平均答案计数统计: 提供不同查询类型的训练、验证和测试集的平均答案数量。
数据集使用
- 数据集加载: 支持加载完整数据集、元信息以及特定查询类型的子部分。
联系方式
- 联系人: Lin Xueyuan
- 邮箱: linxy59@mail2.sysu.edu.cn
引用信息
- 论文引用: 如使用数据集,请引用"TFLEX: Temporal Feature-Logic Embedding Framework for Complex Reasoning over Temporal Knowledge Graph"。
bibtex @inproceedings{ xueyuan2023tflex, title={TFLEX: Temporal Feature-Logic Embedding Framework for Complex Reasoning over Temporal Knowledge Graph}, author={Lin Xueyuan and Haihong E and Chengjin Xu and Gengxian Zhou and Haoran Luo and Tianyi Hu and Fenglong Su and Ningyuan Li and Mingzhi Sun}, booktitle={Thirty-seventh Conference on Neural Information Processing Systems}, year={2023}, url={https://openreview.net/forum?id=oaGdsgB18L} }




