上传者: 42134168
|
上传时间: 2021-11-23 17:30:40
|
文件大小: 18KB
|
文件类型: -
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