pytorch基础和螺旋数据分类学习

代码练习

2.1 pytorch基础

在运行时，遇到了以下报错：

通过了解，@为矩阵乘法，要求运算的元素类型相同，此时m的类型为：

因此改变v的类型即可如下：

2.2 螺旋数据分类

构建线性模型分类：

由上图知，利用该模型的准确率仅为51.4%，对于螺旋形而言线性分类并不准确。

构建两层神经网络分类：

由上图知，准确率为94.3%。做法为添加了ReLU激活函数。