关于02-Logistic Regression/Logistic_Regression.py，这个模型应该只是普通的线性模型？

L1aoXingyu / pytorch-beginner

pytorch tutorial for beginners

3k stars 1.09k forks source link

Closed snjl closed 5 years ago

snjl commented 5 years ago

您好，02-Logistic Regression/Logistic_Regression.py中34行直接将线性的结果输出了，那是否意味着这仅仅是一个线性模型，而非逻辑回归，逻辑回归应该必须有一个激活函数，类似relu或者sigmoid？

L1aoXingyu commented 5 years ago

logistice regression 就是 sigmoid(wx + b)，是一个二分类，而这里是一个10分类，所以sigmoid变成了softmax，softmax被吸收到 nn.CrossEntropyLoss() 里面了。

snjl commented 5 years ago

感谢您的解答！解决了一个困扰我的问题！