构建一个简单的多层前向反馈神经网络以解决 XOR 问题。
- 将第一层设为
Dense()
层,并将节点数设为8,且input_dim
设为 2。 - 在第二层之后使用 softmax 激活函数。
- 将输出层节点设为 2,因为输出只有 2 个类别。
- 在输出层之后使用 softmax 激活函数。
- 对模型运行 10 个 epoch。
Dense()
层,并将节点数设为8,且 input_dim
设为 2。# Build a feed-forward network model = nn.Sequential(nn.Linear(784, 128), nn.ReLU(),