Pytorch_Day02_MNIST数据集识别
欢迎来到黄黄自学室MNIST数据集识别损失函数非线性函数ReLU识别步骤加载数据构建网络模型训练测试加载包utils.py
亲爱的朋友们!
任何时候都要抬头挺胸收下巴,慢慢追赶!
MNIST数据集识别
损失函数
待识别目标【0、1、2、3、4、5、6、7、8、9】
做标签:采用one-hot编码方式
1>=[0,1,0,0,0,0,0,0,0,0]
5>=[0,0,0,0,0,5,0,0,0,0]
def one_hot(label, depth=10):
"""one_hot编码"""
原创
2020-07-01 22:03:41 ·
255 阅读 ·
0 评论