KR-EAR 项目使用教程

1. 项目的目录结构及介绍

KR-EAR 项目的目录结构如下：

KR-EAR/
├── KR-EAR(TransE)/
│   ├── main
│   ├── test
│   └── test_attr
├── KR-EAR(TransR)/
│   ├── main
│   ├── test
│   └── test_attr
├── data/
│   ├── train-rel.txt
│   ├── test-rel.txt
│   ├── train-attr.txt
│   ├── test-attr.txt
│   ├── entity2id.txt
│   ├── relation2id.txt
│   ├── attribute2id.txt
│   └── val2id.txt
├── LICENSE
├── README.md
└── makefile

目录结构介绍

• KR-EAR(TransE)/ 和 KR-EAR(TransR)/: 这两个目录分别包含了基于 TransE 和 TransR 模型的代码。每个目录下有 main 文件用于训练模型，test 文件用于测试模型，test_attr 文件用于测试属性。
• data/: 该目录包含了用于知识库补全任务的数据集。数据集包括训练和测试文件，以及实体、关系、属性和值的映射文件。
• LICENSE: 项目的许可证文件，采用 MIT 许可证。
• README.md: 项目的说明文档，介绍了项目的基本信息和使用方法。
• makefile: 用于编译项目的 Makefile 文件。

2. 项目的启动文件介绍

启动文件

• main: 用于启动训练过程。在 KR-EAR(TransE)/ 或 KR-EAR(TransR)/ 目录下运行 ./main 即可开始训练模型。
• test: 用于启动测试过程。在 KR-EAR(TransE)/ 或 KR-EAR(TransR)/ 目录下运行 ./test 即可开始测试模型。
• test_attr: 用于测试属性。在 KR-EAR(TransE)/ 或 KR-EAR(TransR)/ 目录下运行 ./test_attr 即可开始测试属性。

使用方法

1. 进入相应的模型目录（如 KR-EAR(TransE)/）。
2. 运行 ./main 开始训练模型。
3. 运行 ./test 或 ./test_attr 进行测试。

3. 项目的配置文件介绍

Makefile

项目的 Makefile 文件用于编译项目代码。在项目根目录下运行 make 命令即可编译项目。

数据文件

数据文件位于 data/ 目录下，包括以下文件：

• train-rel.txt: 关系训练文件，格式为 (e1, e2, rel)。
• test-rel.txt: 关系测试文件，格式与 train-rel.txt 相同。
• train-attr.txt: 属性训练文件，格式为 (e1, val, attr)。
• test-attr.txt: 属性测试文件，格式与 train-attr.txt 相同。
• entity2id.txt: 实体及其对应 ID 的映射文件。
• relation2id.txt: 关系及其对应 ID 的映射文件。
• attribute2id.txt: 属性及其对应 ID 的映射文件。
• val2id.txt: 值及其对应 ID 的映射文件。

这些文件是项目运行所必需的数据集，确保它们位于 data/ 目录下，并且格式正确。