【利用PyTorch实现一个简单的三分类】

利用PyTorch实现一个简单的两层全连接神经网络，从而实现三分类

下面的代码使用 PyTorch 库实现了一个简单的两层全连接神经网络，用于解决一个多分类问题。代码首先生成了一个模拟数据集，包含三个类别的数据点。然后定义了一个神经网络 Net 类，该类包含一个隐藏层和一个输出层。ReLU激活函数被用于隐藏层。

在训练过程中，使用随机梯度下降（SGD）作为优化器，并使用交叉熵损失函数来计算网络输出与真实标签之间的误差。在每次迭代过程中，误差通过网络反向传播，优化器根据计算出的梯度更新网络参数。

代码还包含了一个实时绘图部分，以可视化每隔几轮的分类结果和准确率。最后，训练完成后，通过绘制散点图展示了分类器对于这个三分类问题的分类效果。

下面是完整代码：

import os
import torch
import torch.nn.functional as F
from torch.autograd import Variable
import matplotlib.pyplot as plt

os.environ['KMP_DUPLICATE_LIB_OK'] = 'TRUE'

# Fake data
n_data = torch.ones(100, 2)
x0 = torch.normal(2 * n_data, 1)
y0 = torch.zeros(100)
x1 = torch.normal(-2 * n_data, 1)
y1 = torch.ones(100)
x2 = torch.normal(torch.Tensor([[0, -3]] * 100), 1)
y2 = torch.ones(100)