小蝈蝈梦想在于坚持
2022-05-18 23:06
采纳率: 0%
浏览 13

线性神经网络深度学习架构,激活函数的选择

** 我在学习kaggel里房价预测这个数据集时,打算用神经网络来进行预测,如何选择一个好的激活函数或者什么函数可以更好的预测线性结果呢? csdn里面没多少写线性架构的,于是我又去看了做同一个数据集的作品,他使用的激活函数是2层rlue,但是李沐的课程里面又是一个单层的线性层,于是就有了这个问题
**


class Model(nn.Module):
    def __init__(self,in_features,out_features):
        super().__init__()
        self.linear1 = nn.Linear(in_features=in_features,out_features=256,bias=True,dtype=torch.float32)
        self.linear2 = nn.Linear(256,64)
        self.output = nn.Linear(64,out_features= out_features,bias=True,dtype=torch.float32)
 
    def forward(self,x):
        sigma1 = torch.relu(self.linear1(x))
        sigma2 = torch.relu(self.linear2(sigma1))
        output = self.output(sigma2)
        return output
  • 写回答
  • 好问题 提建议
  • 追加酬金
  • 关注问题
  • 邀请回答

1条回答 默认 最新

相关推荐 更多相似问题