李沐深度学习-多项式函数拟合试验

大小猫吃猫饼干

于 2024-01-20 23:44:26 发布

阅读量627

点赞数 6

分类专栏：李沐深度学习编码实现文章标签：深度学习人工智能

本文链接：https://blog.csdn.net/qq_43401942/article/details/135724086

版权

d2lzh_pytorch模块跳转连接

import torch
import numpy as np
import sys

sys.path.append("路径")
import d2lzh_pytorch as d2l

'''
-----------------------------生成人工数据集
样本数n=200
特征数=3
三阶多项式y=1.2x-3.4x^2+5.6x^3+5+ε
'''
n_train, n_test, true_w, true_b = 100, 100, [1.2, -3.4, 5.6], 5
sample_features = torch.randn(n_train + n_test, 1)  # 200x1   单算的一个特征
poly_sample_features = torch.cat((sample_features, torch.pow(sample_features, 2), torch.pow(sample_features, 3)),
                                 dim=1)  # 组合成3个特征
# 因为poly_features取列的元素时，没有对列加[]限制，所以取出来的值不保有原来的维度，而是成为了一维张量，所以labels相应的也是一维张量
labels = true_w[0] * poly_sample_features[:, 0] + true_w[1] * poly_sample_features[: