Note - Pytorch + AllenNLP

2019-08-13  本文已影响0人  汉江岳

AllenNLP是基于pytorch 的NLP研究库,提供一些训练好的模型,对NLP过程中的数据预处理(读写数据集、分词、索引、建词表、minibatching, sorting & padding等)进行了封装,还提供模块化的实验框架,便于NLP研究和开发的快速迭代。

需要关注三部分:

DatasetReader

Model

Trainer

AllenNLP的代码封装比较灵活,便于自定义模型,并且完全基于pytorch,和pytorch搭配可以进一步加速研究和开发效率。

上一篇下一篇

猜你喜欢

热点阅读