pytorch入门学习（六）------ 实现逻辑回归模型

逻辑回归模型的实现

import torch
import torch.nn as nn
import matplotlib.pyplot as plt
import numpy as np
torch.manual_seed(10)

# ============================= step 1/5 生成数据 ========================

sample_nums = 100
mean_value = 1.7
bias = 1
n_data = torch.ones(sample_nums, 2)
x0 = torch.normal(mean_value * n_data, 1) + bias  # 生成正态分布的x训练数据
y0 = torch.zeros(sample_nums)
x1 = torch.normal(-mean_value * n_data, 1) + bias
y1 = torch.ones(sample_nums)

# 将自变量和因变量的数据与标签进行拼接
train_x = torch.cat((x0, x1), 0)
train_y = torch.cat((y0, y1), 0)

# ============================= step 2/5 选择模型 ==============&#