pytorch二分类神经网络

可以使用PyTorch的nn模块来构建二分类神经网络。以下是一个简单的示例代码：

import torch
import torch.nn as nn

class Net(nn.Module):
    def __init__(self):
        super(Net, self).__init__()
        self.fc1 = nn.Linear(2, 5)
        self.fc2 = nn.Linear(5, 1)
        self.sigmoid = nn.Sigmoid()

    def forward(self, x):
        x = self.fc1(x)
        x = self.sigmoid(x)
        x = self.fc2(x)
        x = self.sigmoid(x)
        return x

这个神经网络有两个输入特征和一个输出，使用了一个线性层和一个sigmoid激活函数来构建隐藏层，然后再使用一个线性层和sigmoid激活函数来构建输出层。