Pytorch学习4 - 神经网络一般构建步骤

/ 技术文章 / 0 条评论 / 1560浏览

Pytorch学习4 - 神经网络一般构建步骤

根据前面的例子可以总结出我们构建神经网络的一般步骤,或者说是使用Pytorch构建神经网络的一般套路。

一般构建一个神经网络模型需要下面3大步骤:

  1. 数据预处理:将输入数据经过一些列预处理,使得数据更适用于神经网络的计算,提高神经网络计算的运行效率
  2. 模型构建:构建神经网络模型,通过循环训练调整各个节点的参数和权重,拟合真实数据
  3. 模型验证:验证训练出的模型的准确率和效率,优化调整模型,防止过拟合

数据预处理

  1. 类型变量、类型编码
  1. 数据标准化
  1. 数据切分

模型构建

  1. 构架模型
  1. 建立损失函数和优化器
  1. 训练模型

模型验证

  1. 损失函数下降曲线
  2. 训练时间统计
  3. 预测数据-实际数据对比