NNDL 作业4：第四章课后题

最新推荐文章于 2024-06-09 23:23:33 发布

plum-blossom

最新推荐文章于 2024-06-09 23:23:33 发布

阅读量504

点赞数 1

分类专栏： NNDL实验文章标签：深度学习 pytorch python

本文链接：https://blog.csdn.net/qq_51713698/article/details/127215841

版权

文章目录

习题4-2 试设计一个前馈神经网络来解决XOR问题,要求该前馈神经网络具有两个隐藏神经元和一个输出神经元,并使用ReLU作为激活函数.

分析

XOR输入为2个神经元，输出为一个神经元，此题要求隐藏神经元为2个，故只要两层全连接层即可

数据集：输入为两位二进制，共四种情况，输出为一个二进制表示的数，要么为0，要么为1
网络模型：用pytorch搭建两层全连接层神经网络
训练：计算网络输出，计算损失函数，反向传播，参数更新
求权重和偏置
测试

代码如下：

import torch.nn as nn
import torch
import torch.optim as optim


# 异或门模块由两个全连接层构成
class XORModule(nn.Module):
    def __init__(self):
        super(XORModule, self).__init__()
        self.fc1 = nn.Linear(2, 2)
        self.fc2 = nn.Linear(2, 1)
        self.relu = nn.ReLU()

    def forward(self, x):
        x = x.view(-1,