1. Learn
    2. /
  2. Courses
    3. /
  3. Введение в глубокое обучение с PyTorch
Connected

Exercise

Функции sigmoid и softmax

Функции sigmoid и softmax — ключевые функции активации в глубоком обучении, которые часто используются на финальном шаге нейронной сети.

  • Sigmoid применяется для бинарной классификации
  • Softmax применяется для многоклассовой классификации

Дан тензор выхода сети до применения функции активации. Примените подходящую функцию активации, чтобы получить итоговый результат.

torch.nn уже импортирован как nn.

Instructions 1/2

undefined XP
  • 1
    • Создайте функцию sigmoid и примените её к input_tensor, чтобы получить вероятность для задачи бинарной классификации.
  • 2
    • Создайте функцию softmax и примените её к input_tensor, чтобы получить вероятности для задачи многоклассовой классификации.