4_logistics_regression

logistics

import torch
import torch.nn as nn
from torch.autograd import Variable


class Model(nn.Module):
    def __init__(self):
        super(Model, self).__init__()
        self.linear = nn.Linear(1, 1)

    def forward(self, x):
        y_pred = torch.sigmoid(self.linear(x))
        return y_pred

    def run(self):
        model = Model()
        criterion = nn.BCELoss(reduction='sum')
        opitmizer = torch.optim.SGD(model.parameters(), lr=0.01)

        for epoch in range(1000):
            y_pred = model(x_data)
            loss = criterion(y_pred, y_data)
            print(epoch, loss.item())

            criterion.zero_grad()
            loss.backward()
            opitmizer.step()

        hour_var = Variable(torch.Tensor([[1.0]]))
        print('predict 1 hour:', 1.0, model(hour_var).data[0][0] > 0.5)
        hour_var = Variable(torch.Tensor([[7.0]]))
        print('predict 7 hours:', 7.0, model(hour_var).data[0][0] > 0.5)


if __name__ == "__main__":
    print("Life is short, You need Python!")
    x_data = Variable(torch.Tensor([[1.0], [2.0], [3.0], [4.0]]))
    y_data = Variable(torch.Tensor([[0.], [0.], [1.], [1.]]))
    m = Model()
    m.run()