单层神经网络实现-pytorch-softmax-sigmoid使用
7.示例:单层神经网络实现
本节使用神经网络在iris数据集的多分类示例来说明神经网络的整个流程。一般的神经网络训练包括几个重要的步骤:数据准备,初始化权重,激活函数,前向计算,损失函数,计算损失,反向传播,更新参数,直到收敛或者达到终止条件。
import torch
from torch import sigmoid
import torch.nn.functional as F
import matplotlib.pyplot as plt
from sklearn.datasets impor
原创
2020-08-19 15:14:51 ·
2371 阅读 ·
0 评论