- 博客(1)
- 收藏
- 关注
原创 李宏毅_Machine Learning HW1
loss = criterion(pred, y)#两个一维向量算MSE判断预测结果和真实结果的差距。# 1.1 select feature from the dataset----一般不用。pred = model(x)#在经过模型之后已经变成一个一维向量。#不计算梯度所以不需要 loss.detach().item()early_stop=0#使用最好的模型参数---避免过拟合。optimizer.step()#更新参数。loss.backward()#反向传播。#x:症状 y:阳性与否。
2023-09-27 14:52:53 64 1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人