Skip to content

一个基于transformers的自定义命名实体识别模型示例

Notifications You must be signed in to change notification settings

bqw18744018044/BertForNER

Repository files navigation

一个基于transformers的自定义命名实体识别模型示例

一、简介

该项目是一个基于transformers的命名实体识别模型。主要目标是提供一个代码示例,展示如果基于transformers提供的功能进行快速的模型开发。

二、代码结构

  • data:包含训练数据;
  • config.py:各类参数统一定义在该文件中;
  • data_utils.py:各类数据加载的函数和全局变量定义在该文件中;
  • train_utils.py:训练过程中需要的各种辅助函数或者类定义在该文件中;
  • model.py:模型结构定义在该文件中;
  • train.py:模型训练的流程和入口在该文件中;
  • notebook.ipynb:notebook版本的代码;

三、运行代码

python train.py

四、本项目博客

一个基于transformers的自定义命名实体识别模型示例

About

一个基于transformers的自定义命名实体识别模型示例

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published