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

使用环境：

pytorch：1.1.0

cuda：10.0

gpu：RTX2070

 

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

 

（2）使用DataLoader加载数据

 

（3）定义LSTM模型用于文本二分类

 

（4）定义训练函数和测试函数

 

 

（5）开始模型的训练（并保存最优模型权重），训练较快，2min左右

 

（6）加载模型权重并测试

 

 

 

 

 

 

版权声明：本文为conghuang原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://www.cnblogs.com/conghuang/p/11919094.html