pytorch-tutorial icon indicating copy to clipboard operation
pytorch-tutorial copied to clipboard

代码里展示的是 Softmax Regression,而非 Logistic Regression

Open y1x4 opened this issue 6 years ago • 3 comments

pytorch-tutorial/tutorials/01-basics/logistic_regression/main.py

model = nn.Linear(input_size, num_classes) criterion = nn.CrossEntropyLoss()

y1x4 avatar Oct 10 '19 06:10 y1x4

难道不就是用 softmax regression 嘛 (which is used for multi-class classification problems)

OrangeC93 avatar Nov 20 '19 00:11 OrangeC93

是的,没太看明白,注释里写的是Logistic Regression

yuzi1949 avatar Sep 07 '20 15:09 yuzi1949

看来不是我一个人有疑问

nuaa030710312 avatar Oct 21 '21 07:10 nuaa030710312