Pytorch文本分类(imdb数据集),含DataLoader数据加载,最优模型保存


用pytorch进行文本分类,数据集为keras内置的imdb影评数据(二分类),代码包含六个部分(详见代码)

代码地址为:pytorch-imdb-classification 欢迎star~

使用环境:

pytorch:1.1.0

cuda:10.0

gpu:RTX2070

 

(1)导入相应的库、定义常量以及加载imdb数据

 

(2)使用DataLoader加载数据

 

(3)定义LSTM模型用于文本二分类

 

(4)定义训练函数和测试函数

 

 

(5)开始模型的训练(并保存最优模型权重),训练较快,2min左右

 

(6)加载模型权重并测试

 

 

 

 

 

 


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM