Concise implementation of linear regression based on Pytorch
# 基于 Pytorch 的线性回归简介实现 主要内容:借助 Pytorch 框架,结合 d2l 面向对象设计,完成线性回归的简洁实现 # 准备 1234import numpy as npimport torchfrom torch import nnfrom d2l import torch as d2l torch.nn 包含了常用的神经网络层 # 模型定义 d2l.Module # 参数及其初始化 1234567class LinearRegression(d2l.Module): def __init__(self, lr): super().__init__()...
more...

