第一节课 · 浅谈神经网络

浅谈神经网络

第一节课《浅谈神经网络》配套代码资源：下载后跟着课程一起动手训练你的第一个神经网络——螺旋分类器（Spiral Classifier）。

先让AI干这个！

动手之前，先让 AI 帮你把环境装好。把下面这段话直接发给你的 AI 助手（DeepSeek、ChatGPT、Proma 等都可以），它会自动检测你的电脑配置，用清华镜像源帮你安装适合的 PyTorch：

帮我在我的电脑上安装 PyTorch：
1. 先检测我的操作系统、Python 版本、显卡型号（是否有 NVIDIA GPU 以及 CUDA 支持情况）
2. 使用清华镜像源（-i https://pypi.tuna.tsinghua.edu.cn/simple）安装
3. 根据检测结果选择适合的版本：有 NVIDIA 显卡装 CUDA 版，否则装 CPU 版
4. 给出安装命令并执行
5. 装完后运行 python -c "import torch; print(torch.__version__)" 验证是否成功

为什么让 AI 干？PyTorch 分 CPU 版和 CUDA（GPU）版，选错版本会白装一遍；AI 会自动检测你的电脑配置并选对版本，清华镜像源还能避开官方源下载慢的问题。装完记得解压代码资源，用 pip install -r requirements.txt 安装剩余依赖。

资源内容

  • train.py训练脚本：训练一个两层的浅层神经网络分类器
  • train.ipynbJupyter Notebook 教学版，带完整讲解与可视化
  • spiral_classifier.py已训练好的模型推理脚本
  • test_spiral_classifier.py模型测试脚本
  • spiral_classifier.pt训练好的模型权重文件
  • requirements.txt依赖清单（torch、matplotlib、numpy 等）
  • *.png数据与结果可视化图片

使用方法

解压后，在终端中进入目录并安装依赖：

cd 01-浅谈神经网络
pip install -r requirements.txt

运行训练脚本：

python train.py

下载代码资源

下载即将开始；如果浏览器未自动开始，请点击下方按钮。

手动下载代码资源