区分类型 (分类)

学习资料:

• 本节的全部代码
• 用 Tensorflow 达到同样效果的代码
• 我制作的 什么是神经网络 动画简介
• PyTorch 官网

要点¶

这次我们也是用最简单的途径来看看神经网络是怎么进行事物的分类.

建立数据集¶

我们创建一些假数据来模拟真实的情况. 比如两个二次分布的数据, 不过他们的均值都不一样.

建立神经网络¶

建立一个神经网络我们可以直接运用 torch 中的体系. 先定义所有的层属性(__init__()), 然后再一层层搭建(forward(x))层于层的关系链接. 这个和我们在前面 regression 的时候的神经网络基本没差. 建立关系的时候, 我们会用到激励函数, 如果还不清楚激励函数用途的同学, 这里有非常好的一篇动画教程.

训练网络¶

训练的步骤很简单, 如下:

可视化训练过程¶

为了可视化整个训练的过程, 更好的理解是如何训练, 我们如下操作:

所以这也就是在我 github 代码 中的每一步的意义啦.