A puzzle about the model in chapter 2.

L1aoXingyu / pytorch-beginner

pytorch tutorial for beginners

3k stars 1.09k forks source link

A puzzle about the model in chapter 2. #18

Open xzhws opened 5 years ago

xzhws commented 5 years ago

The model in chapter 2 should be multilayer perceptron instead of logistic regression, since which can only be used in two classfication problem. How about changing the model to softmax which is available for multi-classification problem? thanks.

L1aoXingyu commented 5 years ago

You are right, theoretically logistic regression deal with two-class classification problem and I will think about change this example to a two-class problem.