稍微了解了一下机器学习的相关内容
对这种新的解决问题的方法很感兴趣
想亲自验证一番
于是呢,花了点时间,搭建了一下环境
然后最简单而又不失神秘感的当属函数拟合实验了
不需要准备什么数据,省时间
于是拼凑式编程,搞了几行代码
import torch
import matplotlib.pyplot as plt
sum_train=1000
batch_size=10
learn_rate=0.001
lim_loss=0.0001
def makedata():
x=torch.unsqueeze(torch.linspace(0, 1, 1000),dim=1)
y=x**2-x+0.25
return x,y
#