深度学习架构师机考代码题库.docxVIP

深度学习架构师机考代码题库

1，搭建一个可用于波士顿房价预测的全连接网络模型（可借助PyTorch）；

importtorch

fromtorchimportnn

classModel(nn.Module):

def__init__(self,n_features=13):

super(Model,self).__init__()

self.linear1=nn.Linear(in_features=n_features,out_features=8)

self.linear2=nn.Linear(in_features=8,out_features=1)

defforward(self,x):

x=self.linear1(x)

x=torch.relu(x)

x=self.linear2(x)

returnx

2，搭建一个可用于手写数字识别的卷积网络模型（黑白图像，28*28大小，10个分类；可借助PyTorch）；

importtorch

fromtorchimportnn

classModel(nn.Module):

def__init__(s