pytorch 实现简单二元分类器以及可视化

本文转载于 SegmentFault 社区作者：zack这个文章是用 pytorch 和 matplotlib 实现一个二元分类器并且可视化。思路：自己生成两团数据定义自己的神经网络类训练网络打印出边界先放效果图：  关于可视化定义网络、训练网络主要没什么好说的啦其实，毕竟有 pytorch 这么好的框架，已经提供了如此简单的代码工作。主要是可视化的技巧。主要是 matplotlib 中有个 contourf，本身是画等高线用的，就是地理中那个三维图投射到二维图的那种图。我们可以把这个用到可视化上来（当然只是3维的，如果是更高维度就没法用这个可视化了）。 具体怎么可视化的?首先，先自己生成 200 个训练数据（这步对应 getData 函数），然后把属于不同类别的数据染上不同颜色；然后，进行网络的训 ………………………………