Začněte nyníZačněte zdarma

Vytvoření binárního klasifikátoru v PyTorch

Malá neuronová síť s jednou lineární vrstvou a sigmoidovou funkcí funguje jako binární klasifikátor – chová se stejně jako logistická regrese.

Procvič si sestavení této jednoduché sítě a interpretaci výstupu klasifikátoru.

Toto cvičení je součástí kurzu

Introduction to Deep Learning with PyTorch

Zobrazit kurz

Interaktivní cvičení na vyzkoušení si v praxi

Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.

import torch
import torch.nn as nn

input_tensor = torch.Tensor([[3, 4, 6, 2, 3, 6, 8, 9]])

# Implement a small neural network for binary classification
model = nn.Sequential(
  nn.____(____),
  nn.____()
)

output = model(input_tensor)
print(output)
Upravit a spustit kód