1.线性模型输入输出
线性模型的输入是多维特征的样本,每个样本对应一个输出。可以抽象地理解为一个同学是一个样本,每个同学测试“身高、体重、视力、色盲”等身体状况,根据这些特征判断该同学是否健康,健康为1,不健康为0。


2.torch.nn.Linear线性模块
基础线性模型的网络架构如图所示,训练目的就是优化权重参数(w和b)使得损失loss最小,损失函数的计算公式可以根据需要选择。

如果是二分类问题,那么输出应该是属于某一类的概率,需要使用非线性函数(sigmoid函数)将y映射到[0,1]之间,且非线性函数可以提高模型的非线性学习能力,能够更好地拟合模型。

模型进一步演变为下图所示,使用

最低0.47元/天 解锁文章
5777

被折叠的 条评论
为什么被折叠?



