登录 注册
当前位置:主页 > 资源下载 > 32 > LSTM-CRF在PyTorch中的实现:lstm-crf-pytorch

LSTM-CRF在PyTorch中的实现:lstm-crf-pytorch

  • 更新:2024-05-19 16:09:20
  • 大小:18KB
  • 推荐:★★★★★
  • 来源:网友上传分享
  • 类别:其它 - 开发技术
  • 格式:ZIP

资源介绍

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