pytorch_task3

PyTorch实现Logistic regression
PyTorch基础实现代码
用PyTorch类实现Logistic regression,
torch.nn.module写网络结构

首先制造假数据,在plt中显示分布,从中看出分为两类,这时需要设计一个模型将其分类

在这里插入图片描述
引入nn

在这里插入图片描述
在这里插入图片描述

在这里插入图片描述
在这里插入图片描述

在这里插入图片描述
显示出训练结果,以及从训练迭代中看出loss在逐渐减小,而由于最开始生成的假数据为两类,所以精度一直为100

猜你喜欢

转载自blog.csdn.net/qq_42017043/article/details/89196573