AI学习我每天都在进步
LIQIANDI
这个作者很懒,什么都没留下…
展开
-
使用pytorch实现线性回归
线性回归都是包括以下几个步骤:定义模型、选择损失函数、选择优化函数、训练数据、测试 import torch # 构建数据集 x_data= torch.Tensor([[1.0],[2.0],[3.0],[4.0],[5.0],[6.0]]) y_data= torch.Tensor([[2.0],[4.0],[6.0],[8.0],[10.0],[12.0]]) #定义模型 class LinearModel(torch.nn.Module): def __init__(self): .原创 2020-06-09 17:02:50 · 683 阅读 · 0 评论 -
以梯度下降法实现简单线性回归
(跟着B站“刘二大人”pytorch学习第二课,实现一个简单的线性函数,给出三组数据值,预测第四组的值。) 1、梯度下降算法:通过迭代找到目标函数的最小值,或者收敛到最小值 (梯度下降算法不一定能得到最优解,但是能得到局部最优,满足大部分需求。其中a为学习率,取得值要小一些,可以处理出现鞍点的情况) 由 得到 #使用梯度下降算法 import numpy as np import torch import matplotlib.pyplot as plt w=1.0 # 定义数据集 x_..原创 2020-06-09 09:52:50 · 1196 阅读 · 0 评论 -
线性模型/使用matplotlib绘制3D图像
跟着刘老师:https://www.bilibili.com/video/av92862340学习的pytorch课程,第一节课课后习题: 使用pytorch实现一个简单的线性模型,并调用matplotlib输出模型图像。 (本菜鸡注释真的多...) import torch import numpy as np import matplotlib.pyplot as plt #绘图用的模块 from mpl_toolkits.mplot3d import Axes3D #绘制3D坐标的函数 x_.原创 2020-06-09 01:18:37 · 716 阅读 · 1 评论