PyTorch实现Logistic regression
- PyTorch基础实现代码
- 用PyTorch类实现Logistic regression, torch.nn.module写网络结构
Logistic regression 简单的理解为输出值在0到1之间的回归,可用其作为概率来分析判断
一般用Sigmoid函数,实现普通值到0和1之间的值的映射
所以 Logistic regression 的数学表达式如下
2.Pytorch实现
import torch
from torch.autograd import Variable
x_data = Variable(torch.Tensor([[0.6], [1.0], [3.5], [4.0]]))
y_data = Variable(torch.Tensor([[0.], [0.], [1.], [1.]]))
class Model(torch.nn.Module):
def __init__(self):
super(Model, s