代码练习系列（一）——搭建一个可用于波士顿房价预测的全连接网络模型

马斯洛金字塔下的小灵猴儿

已于 2024-07-15 20:22:02 修改

阅读量138

点赞数 1

分类专栏：深度学习架构师考试通关秘籍文章标签：深度学习人工智能考试

于 2024-07-15 18:32:00 首次发布

本文链接：https://blog.csdn.net/GOOD_CODER/article/details/140446054

版权

深度学习架构师考试通关秘籍专栏收录该内容

35 篇文章 1 订阅 ¥19.90 ¥99.00

订阅专栏

超级会员免费看

import torch
from torch import nn

class Model(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)
    
    def forward(self, x):
        x = self.linear1(x)
        x = torch.relu(x)
        x = self.linear2(x)
        return x