AutoKG 项目使用教程

AutoKG 项目使用教程

AutoKG [WWWJ 2024] LLMs for Knowledge Graph Construction and Reasoning: Recent Capabilities and Future Opportunities 项目地址: https://gitcode.com/gh_mirrors/aut/AutoKG

1. 项目目录结构及介绍

AutoKG 项目是一个用于知识图谱构建和推理的开源项目，其目录结构如下：

AutoKG/
├── KG Construction/            # 知识图谱构建相关文件
│   ├── DuIE2.0/
│   │   ├── datas/              # DuIE2.0 数据集
│   │   ├── prompts/            # DuIE2.0 的提示文件
│   │   ├── duie_processor.py   # DuIE2.0 数据预处理脚本
│   │   └── duie_prompts.py     # DuIE2.0 提示生成脚本
│   ├── MAVEN/
│   │   ├── datas/              # MAVEN 数据集
│   │   ├── prompts/            # MAVEN 的提示文件
│   │   ├── maven_processor.py  # MAVEN 数据预处理脚本
│   │   └── maven_prompts.py    # MAVEN 提示生成脚本
│   ├── RE-TACRED/
│   │   ├── datas/              # RE-TACRED 数据集
│   │   ├── prompts/            # RE-TACRED 的提示文件
│   │   ├── retacred_processor.py # RE-TACRED 数据预处理脚本
│   │   └── retacred_prompts.py # RE-TACRED 提示生成脚本
│   └── SciERC/
│       ├── datas/              # SciERC 数据集
│       ├── prompts/            # SciERC 的提示文件
│       ├── scierc_processor.py # SciERC 数据预处理脚本
│       └── scierc_prompts.py   # SciERC 提示生成脚本
├── KG Reasoning/               # 知识图谱推理相关文件
│   ├── FB15k-237/
│   │   ├── data/               # FB15k-237 数据样本
│   │   ├── prompts/            # FB15k-237 的提示文件
│   │   └── system_eval/        # ATOMIC2020 系统评估
│   └── ATOMIC2020/
│       ├── data/               # ATOMIC2020 数据样本
│       └── prompts/            # ATOMIC2020 的提示文件
├── Virtual Knowledge Extraction/ # 虚拟知识提取相关文件
├── README.md                   # 项目说明文件
└── AutoKG.py                   # 项目主启动文件

2. 项目的启动文件介绍

项目的启动文件为 AutoKG.py，该脚本负责初始化和启动整个 AutoKG 系统。在运行此脚本之前，确保已经正确配置了相关环境变量和 API 密钥。

cd AutoKG
python AutoKG.py

3. 项目的配置文件介绍

项目的配置文件主要集中在各个子模块的提示文件（.py）中，例如 duie_prompts.py、maven_prompts.py、retacred_prompts.py 和 scierc_prompts.py。这些文件中定义了用于训练和推理的提示（prompts），以及相关的预处理参数。

此外，主启动文件 AutoKG.py 中也可能包含一些配置参数，如 API 密钥等。这些参数通常位于文件的顶部，并在程序运行时被读取和设置。

请根据实际需要调整这些配置文件中的参数，以确保项目的正确运行。

AutoKG [WWWJ 2024] LLMs for Knowledge Graph Construction and Reasoning: Recent Capabilities and Future Opportunities 项目地址: https://gitcode.com/gh_mirrors/aut/AutoKG