【文件属性】:
文件名称:bi-lstm-crf:BI-LSTM-CRF模型的PyTorch实现
文件大小:16KB
文件格式:ZIP
更新时间:2021-05-07 15:52:29
nlp crf pytorch ner word-segmentation
BI-LSTM-CRF模型的PyTorch实现。
特征:
与相比,执行了以下改进:
全面支持小批量计算
完全矢量化的实现。 特别是,删除了“得分句”算法中的所有循环,从而极大地提高了训练效果
支持CUDA
用于非常简单的API
START / STOP标签会自动添加到CRF中
包含一个内部线性层,该线性层可从要素空间转换为标签空间
专门用于NLP序列标记任务
轻松训练自己的序列标记模型
麻省理工学院执照
安装
依存关系
的Python 3
安装$ pip install bi-lstm-crf
训练
语料库
以指定的准备语料库
也有一个示例语料库
训练
$ python -m bi_lstm_crf corpus_dir --model_dir " model_xxx "
更多
训练曲线
import pandas as pd
import matplotlib . pyplot
【文件预览】:
bi-lstm-crf-master
----MANIFEST.in(0B)
----LICENSE(1KB)
----bi_lstm_crf()
--------model()
--------__init__.py(76B)
--------__main__.py(36B)
--------app()
----setup.py(1KB)
----README.md(3KB)
----.gitignore(1KB)