CogKR: 认知知识图关系学习的源代码与数据集-一键式推理

简介：
CogKR提供了一种新颖的方法来学习认知知识图中的关系，包括源代码和数据集。它支持一键式推理，简化了复杂的知识处理流程，使用户能够轻松地从中获取有价值的信息。 CogKR 是一种一键式关系学习的认知知识图推理系统（杜正晓、周常、丁铭、杨红霞、唐洁正在开发）。其先决条件包括 Python 3 和 PyTorch 版本 >=1.1.0，以及 NVIDIA GPU 加上 CUDA 和 cuDNN。安装步骤如下：克隆此仓库使用命令 `git clone https://github.com/THUDM/CogKR` ，然后进入目录 `cd CogKR` 并通过运行命令 `pip install -r requirements.txt` 安装依赖项。 在实验中，我们采用了两个公共数据集 NELL-One 和 Wiki-One（经过一些修改）。原始和预处理的数据可以从相应的资源下载并解压缩到指定的文件夹内。如需使用自定义数据集，请参考相关文档中的“使用数据集”部分进行操作。 对于训练过程，只需按照上述说明准备环境与数据即可开始。