深度学习练习:利用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。
原创
2023-10-13 15:01:57 ·
94 阅读 ·
0 评论