import torchvision
train_set = torchvision.datasets.MNIST(root='../dataset/mnist', train=True, download=True)
test_set = torchvision.datasets.MNIST(root='../dataset/mnist', train=False, download=True)
##如果事先已经下载好数据集,则download=False
Regression vs Classification
- 二分类问题只需要输出一个概率,因为两个概率相加等于1
- 在线性模型中,
,是个实数;在分类问题中,
,是个概率
- Logistic Function:
,可以使得值介于0~1之间
- Loss Function for Linear Regression:
- Loss Function for Binary Classification: