構建一個簡單的多層前向反饋神經網絡以解決 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(),