每天五分钟深度学习PyTorch:基于全连接神经网络完成手写字体识别
本文重点
上一节我们学习了搭建普通的全连接神经网络,我们现在用它来解决一个实际问题,我们用它跑一下手写字体识别的数据,然后看看它的效果如何。
网络模型
class ThreeNet(nn.Module) :
def __init__ (self,in_dim,n_hidden_1,n_hidden_2,out_dim):
super(ThreeNet, self).__init__()
self.layer1 = nn.Linear (in_dim, n_hidden_1)
self.layer2 = nn.Linear(n_hidden_1,n_hidden_2)
self.layer3 = nn.Linear(n_hidden_2, out_dim)
<def forward(self,x):
x =self.layer1(x)
x =self.layer2(x)
x =self.layer3(x)
return x