NLP-基于NER的实体识别模型

阅读量：

说明

模型：BiLSTM结合CRF
框架：tensorflow 1.14.0
硬件：RTX 2080Ti（11G）
其他：使用BIO标注方式
在命名实体识别领域，本人尚无实际操作经验，代码中可能存在不妥之处，敬请各位予以指正

代码

目录结构如下:

NER
- dataset
  - labels.txt
  - train_data
- models
- datautil.py
- ner.py
- train.py

ner.py

复制代码

    # -*- coding:utf-8 -*-
    import os
    from tqdm import tqdm
    import tensorflow as tf
    from tensorflow.contrib import crf
    
    # 设置输出级别
    tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.INFO)
    
    
    def get_key(kwargs: dict, key: str, default):
    """
    获取key的值
    :param kwargs

全部评论 (0)

还没有任何评论哟~