-
LSTM-CRF在PyTorch中的实现:lstm-crf-pytorch
资源介绍
PyTorch中的LSTM-CRF
用于序列标记的双向LSTM-CRF的最小PyTorch(1.7.1)实现。
支持的功能:
CUDA的小批量培训
嵌入层中的查找,CNN,RNN和/或自我关注
分层递归编码(HRE)
条件随机场(CRF)的PyTorch实现
CRF损失的矢量化计算
矢量化维特比解码
用法
培训数据的格式应如下:
token/tag token/tag token/tag ...
token/tag token/tag token/tag ...
...
有关更多详细信息,请参见每个子目录中的README.md。
准备数据:
python3 prepare.py training_data
训练:
python3 train.py model char_to_idx word_to_idx tag_to_idx training_data.csv (v
- 上一篇: botorch:PyTorch中的贝叶斯优化
- 下一篇: ABAQUS软土路基沉降分析