PyTorch笔记3-分类

本系列笔记为莫烦PyTorch视频教程笔记 github源码

概要

用 PyTorch 建立神经网络对事物进行分类

import torch
from torch.autograd import Variable
import torch.nn.functional as F         # activation function
import matplotlib.pyplot as plt

建立数据集

创建一些假数据来模拟真实的情况，比如两个二次分布的数据，不过他们的mean（均值）不一样。