deepke/example/ner/standard

快速上手

环境依赖

python >= 3.7

• pytorch-transformers==1.2.0
• torch==1.2.0
• seqeval==0.0.5
• tqdm==4.31.1
• nltk==3.4.5

克隆代码

git clone git@github.com:zjunlp/DeepKE.git

使用pip安装

首先创建python虚拟环境，再进入虚拟环境

• 安装依赖：pip install -r requirements.txt

使用数据进行训练预测

• 存放数据：在data文件夹下存放数据集。主要有三个文件：

  • train.txt：存放训练数据集
  • valid.txt：存放验证数据集
  • test.txt：存放测试数据集

• 开始训练：python run.py (训练所用到参数都在conf文件夹中，修改即可)

• 每次训练的日志保存在 logs 文件夹内，模型结果保存在 checkpoints 文件夹内。

• 进行预测 python predict.py

模型内容

BERT