PyTorch实现Logistic regression
PyTorch基础实现代码
用PyTorch类实现Logistic regression,
torch.nn.module写网络结构
首先制造假数据,在plt中显示分布,从中看出分为两类,这时需要设计一个模型将其分类
引入nn
显示出训练结果,以及从训练迭代中看出loss在逐渐减小,而由于最开始生成的假数据为两类,所以精度一直为100
PyTorch实现Logistic regression
PyTorch基础实现代码
用PyTorch类实现Logistic regression,
torch.nn.module写网络结构
首先制造假数据,在plt中显示分布,从中看出分为两类,这时需要设计一个模型将其分类
引入nn
显示出训练结果,以及从训练迭代中看出loss在逐渐减小,而由于最开始生成的假数据为两类,所以精度一直为100