Pytorch 6

罗切斯特回归模子

加了激活函数
加了激活函数之后类

2. class LogisticRegressionModel(torch.nn.Module):
3.     def __init__(self):
4.         super(LogisticRegressionModel, self).__init__()
5.         self.linear = torch.nn.Linear(1,1)
7.     def forward(self, x):
8.         # y_pred = F.sigmoid(self.linear(x))
9.         y_pred = torch.sigmoid(self.linear(x))
10.         return y_pred

复制代码

这里以sigmoid激活函数为例，具体的激活函数还是查资料
丧失函数

2. criterion = torch.nn.BCELoss(size_average = False)

复制代码

也是接受一个参数，是否求均匀值
求均匀值会影响到dw和db是否要乘1/n，会影响到学习率的设置，如果求均值则偏导也有1/n


免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。