零基础入门Pytorch框架(一)--实现文本分类

1. 基于pytorch实现快速搭建神经网络

首先使用pytorch框架实现全连接层网络,一般搭建神经网络的步骤如下:

  • 数据集获取处理
  • 定义神经网络
  • 定义损失函数和优化器
  • 训练模型
  • 预测模型

1.1 数据集获取处理

一个简单的图像分类问题,输入图像的高和宽均为2像素,色彩为灰度

torch.manual_seed(7)
X =torch.rand((7,2,2))
target = torch.randint(0,2,(7,))

零基础入门Pytorch框架(一)--实现文本分类_第1张图片

你可能感兴趣的:(算法工程师工具,pytorch,分类,深度学习)