深度学习之PyTorch---- Logistic回归（二分类问题）

"""
Logistic 回归
"""


class LogisticRegression(nn.Module):
    def __init__(self):
        super(LogisticRegression,self).__init__()
        self.lr = nn.Linear(2,1)
        self.sm = nn.Sigmoid()
    
    def forward(self,x):
        x = self.lr(x)
        x = self.sm(x)
        return x


if __name__ == '__main__':
    with open('data.txt' , 'r') as f:
        data_list = f.readlines()
        data_list = [i.split('\n')[0] for i in data_list]
        data_list = [i.split(',') for i in data_list]
        data = [(float(i[0]),float(i[1]),float(i[2])) for i in data_list]
        data = torch.Tensor(data)

    
    logistic_model = LogisticRegression()

    # 定义损失函数和优化器
    criterion = nn.BCELoss()
    optimizer = optim.SGD(logistic_model.parameters(),lr=1e-3,momentum=0.9)

    start = time.time()
    for epoch