机器学习pytorch平台代码学习笔记（5）——分类模型