zetxie的博客

原创 深度学习练习：利用softmax回归实现FashionMNIST数据集分类

批次数#输入尺寸(28*28)#输出尺寸(10个类别)#学习率lr = 0.01#训练轮数#返回的是DataLoader对象softmax(x)#如果预测值与真实值相同，那么cmp对应位置就位true，否则为false#返回预测正确的个数#正确预测的数量#总的样本数量#总的损失值else :#返回平均损失值和准确率#使用W,b作为权重print(l)print(ac)plt.show()#判断m是否为linear层#这里不需要初始化bais，默认为0。