ConMask 项目使用教程

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

ConMask 项目的目录结构如下：

ConMask/
├── data/
│   └── dbpedia50/
├── ndkgc/
│   ├── models/
│   ├── ops/
│   └── tools/
├── LICENSE
├── README.md
└── ...

目录介绍

• data/: 存放数据集的目录，例如 dbpedia50 数据集。
• ndkgc/: 核心代码目录，包含模型、操作和工具。
  • models/: 存放模型定义文件。
  • ops/: 存放自定义操作文件，例如负采样操作。
  • tools/: 存放辅助工具文件。
• LICENSE: 项目许可证文件。
• README.md: 项目说明文件。

2. 项目的启动文件介绍

ConMask 项目的启动文件主要是通过命令行执行的 Python 脚本。以下是启动文件的介绍：

启动命令

python3 -m ndkgc models fcn_model_v2 checkpoint_db50_v2_dr_uniweight_2 data/dbpedia50 --force_eval --layer 3 --conv 2 --lr 1e-2 --keep_prob 0.5 --max_content 512 --pos 1 --neg 4 --noopen --neval 5000 --eval --nofilter

命令参数介绍

• ndkgc: 模块名称。
• models: 子模块名称。
• fcn_model_v2: 模型名称。
• checkpoint_db50_v2_dr_uniweight_2: 检查点文件。
• data/dbpedia50: 数据集路径。
• --force_eval: 强制评估。
• --layer 3: 设置层数。
• --conv 2: 设置卷积层数。
• --lr 1e-2: 设置学习率。
• --keep_prob 0.5: 设置保留概率。
• --max_content 512: 设置最大内容长度。
• --pos 1: 设置正样本数量。
• --neg 4: 设置负样本数量。
• --noopen: 关闭开放世界评估。
• --neval 5000: 设置评估次数。
• --eval: 执行评估。
• --nofilter: 不使用过滤器。

3. 项目的配置文件介绍

ConMask 项目没有显式的配置文件，但可以通过命令行参数进行配置。以下是一些常用的配置参数：

常用配置参数

• --layer: 设置模型层数。
• --conv: 设置卷积层数。
• --lr: 设置学习率。
• --keep_prob: 设置保留概率。
• --max_content: 设置最大内容长度。
• --pos: 设置正样本数量。
• --neg: 设置负样本数量。
• --noopen: 关闭开放世界评估。
• --neval: 设置评估次数。
• --eval: 执行评估。
• --nofilter: 不使用过滤器。

通过这些命令行参数，可以灵活地配置和启动 ConMask 项目。