单隐含层改成双隐含层的代码
时间: 2023-08-28 07:07:47 浏览: 54
好的,以下是单隐含层改成双隐含层的代码,代码中用到了PyTorch:
```python
import torch
import torch.nn as nn
class DoubleHiddenLayerModel(nn.Module):
def __init__(self, input_size, hidden_size_1, hidden_size_2, output_size):
super(DoubleHiddenLayerModel, self).__init__()
self.fc1 = nn.Linear(input_size, hidden_size_1)
self.fc2 = nn.Linear(hidden_size_1, hidden_size_2)
self.fc3 = nn.Linear(hidden_size_2, output_size)
self.relu = nn.ReLU()
def forward(self, x):
out = self.fc1(x)
out = self.relu(out)
out = self.fc2(out)
out = self.relu(out)
out = self.fc3(out)
return out
```
这个双隐含层模型有两个隐藏层,分别是 `fc1` 和 `fc2`,每个隐藏层后面都跟着一个 ReLU 激活函数。`fc3` 是输出层。和单隐含层模型不同,这里我们需要三个线性层来连接输入、两个隐藏层和输出。
如果需要修改隐藏层的大小,只需要修改 `hidden_size_1` 和 `hidden_size_2` 的值即可。