Note - Pytorch + AllenNLP
2019-08-13 本文已影响0人
汉江岳
AllenNLP是基于pytorch 的NLP研究库,提供一些训练好的模型,对NLP过程中的数据预处理(读写数据集、分词、索引、建词表、minibatching, sorting & padding等)进行了封装,还提供模块化的实验框架,便于NLP研究和开发的快速迭代。
需要关注三部分:
DatasetReader
Model
Trainer
AllenNLP的代码封装比较灵活,便于自定义模型,并且完全基于pytorch,和pytorch搭配可以进一步加速研究和开发效率。