DRAGON 项目使用教程

DRAGON 项目使用教程

dragon [NeurIPS 2022] DRAGON 🐲: Deep Bidirectional Language-Knowledge Graph Pretraining 项目地址: https://gitcode.com/gh_mirrors/dra/dragon

1. 项目目录结构及介绍

DRAGON 项目的目录结构如下：

dragon/
├── figs/
├── modeling/
├── preprocess_utils/
├── scripts/
├── utils/
├── utils_biomed/
├── .gitignore
├── LICENSE
├── README.md
├── download_raw_data.sh
├── dragon.py
├── preprocess.py

目录介绍

• figs/: 存放项目相关的图表文件。
• modeling/: 包含模型的实现代码。
• preprocess_utils/: 数据预处理的工具代码。
• scripts/: 包含项目的启动脚本和训练、评估脚本。
• utils/: 通用工具函数和辅助代码。
• utils_biomed/: 生物医学领域的专用工具代码。
• .gitignore: Git 忽略文件配置。
• LICENSE: 项目许可证文件。
• README.md: 项目介绍和使用说明。
• download_raw_data.sh: 下载原始数据的脚本。
• dragon.py: 项目的主文件，包含主要的逻辑和功能。
• preprocess.py: 数据预处理的脚本。

2. 项目启动文件介绍

项目的启动文件主要是 dragon.py 和 preprocess.py。

dragon.py

dragon.py 是项目的主文件，包含了主要的逻辑和功能。它负责加载模型、处理输入数据、进行推理和输出结果。

preprocess.py

preprocess.py 是数据预处理的脚本，负责将原始数据转换为模型可以处理的格式。它包括数据清洗、特征提取、子图提取等功能。

3. 项目的配置文件介绍

DRAGON 项目没有明确的配置文件，但可以通过命令行参数或环境变量来配置项目的行为。主要的配置项包括：

• CUDA_VISIBLE_DEVICES: 指定使用的 GPU 设备。
• num_processes: 指定数据预处理时的并行进程数。
• run: 指定要运行的任务类型，如 common, csqa, obqa, medqa 等。

示例配置

CUDA_VISIBLE_DEVICES=0 python preprocess.py -p 4 --run common csqa obqa

以上命令表示使用 GPU 0 进行数据预处理，并行进程数为 4，运行 common, csqa, obqa 三个任务。

---

通过以上步骤，您可以了解 DRAGON 项目的目录结构、启动文件和配置方法，从而更好地使用和开发该项目。

dragon [NeurIPS 2022] DRAGON 🐲: Deep Bidirectional Language-Knowledge Graph Pretraining 项目地址: https://gitcode.com/gh_mirrors/dra/dragon