用pytorch进行文本分类,数据集为keras内置的imdb影评数据(二分类),代码包含六个部分(详见代码)使用环境:pytorch:1.1.0cuda:10.0gpu:RTX2070(1)导入相应的库、定义常量以及加载imdb数据(2)使用DataLoader加载数据(3)定义LSTM模型用于文本二分类(4)定义训练函数和测试函数(5)开始模型的训练(并保存最优模型权重),训练较快,2min左右(6)加载模型权重并测试