BP神经网络实现分类模型

训练数据生成

每个x向量维度为3,y标签使用one-hot编码进行3分类。

生成的数据如下:

构建BP神经网络模型


初始化权重和偏置矩阵如下:

训练神经网络


绘制梯度下降损失函数曲线图

附(如果采用梯度下降更新权重,即权重值发生少量变化后,基于损失值的变化更新权重的方法如下:

posted @ 2024-05-21 23:34  Sherioc  阅读(86)  评论(0)    收藏  举报