pytorch

利用PyTorch 实现一个简单二分类器

Pytorch是一个深度学习框架,可以 轻松地构建神经网络模型 ,应用主流的算法如LR、CNN等。安装非常简单。本文利用sklearn生成的数据,构建1个简单的神经网络,实现分类。 准备工作: 1、安装numpy matplotlib scikit-learn pytorch 命令行:pip install numpy matplotlib scikit-learn torch 2、完整代码如下: 3、输出结果如下,效果还是不错的。