1. Learn
    2. /
  2. 课程
    3. /
  3. 使用 PyTorch 的深度学习入门
Connected

道练习

在 PyTorch 中构建二分类器

回顾一下：一个包含单个线性层、后接 sigmoid 函数的小型神经网络就是一个二分类器。它的行为与逻辑回归相同。

练习搭建这个小型网络，并解读分类器的输出。

说明 1 / 共 2 个

undefined XP
    1
    2
  • 创建一个神经网络，输入为 1x8 张量，输出为单个用于二分类的数值。
  • 将线性层的输出传入 sigmoid，以得到概率。