⽤ nn.Module 实现⼀个逻辑回归模型

好的，我可以为您演示如何使用 PyTorch 中的 nn.Module 实现一个逻辑回归模型。逻辑回归是一种二分类模型，我们可以使用 sigmoid 激活函数将输出值压缩到 [0, 1] 的范围内，表示样本属于某一类的概率。

以下是一个简单的逻辑回归模型的示例代码：

import torch
import torch.nn as nn

class LogisticRegression(nn.Module):
    def __init__(self, input_size):
        super(LogisticRegression, self).__init__()
        self.linear = nn.Linear(input_size, 1)
        self.sigmoid = nn.Sigmoid()

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

上述代码中，我们首先定义了一个继承自 nn.Module 的类 LogisticRegression，并在其中实现了构造函数和前向传播函数。构造函数中，我们定义了一个线性层（nn.Linear）作为模型的主要计算模块，并通过 nn.Sigmoid() 定义了一个 sigmoid 激活函数用于输出。

在前向传播函数中，我们首先将输入数据 x 传入线性层进行计算，然后通过 sigmoid 激活函数将结果压缩到 [0, 1] 的范围内，最终输出分类结果。

最后，我们可以通过以下代码使用该模型对数据进行分类：

# 定义模型
model = LogisticRegression(input_size=2)

# 定义损失函数和优化器
criterion = nn.BCELoss()
optimizer = torch.optim.SGD(model.parameters(), lr=0.01)

# 训练模型
for epoch in range(num_epochs):
    # 前向传播
    outputs = model(inputs)
    loss = criterion(outputs, labels)
    
    # 反向传播和优化
    optimizer.zero_grad()
    loss.backward()
    optimizer.step()

    # 打印日志
    if (epoch+1) % 100 == 0:
        print('Epoch [{}/{}], Loss: {:.4f}'.format(epoch+1, num_epochs, loss.item()))

# 测试模型
with torch.no_grad():
    predicted = model(inputs).round()
    accuracy = (predicted == labels).sum().item() / len(labels)
    print('Accuracy: {:.2f}%'.format(accuracy * 100))

其中，我们首先定义了一个输入大小为 2 的逻辑回归模型，并使用二元交叉熵损失函数（nn.BCELoss）和随机梯度下降优化器（torch.optim.SGD）进行训练。训练完成后，我们使用该模型对数据进行测试，并计算分类精度。