Aan de slagBegin gratis

Een binaire classifier bouwen in PyTorch

Onthoud dat een klein neuraal netwerk met één lineaire laag gevolgd door een sigmoid-functie een binaire classifier is. Het werkt net als een logistische regressie.

Oefen met het bouwen van dit kleine netwerk en het interpreteren van de output van de classifier.

Deze oefening maakt deel uit van de cursus

Introductie tot Deep Learning met PyTorch

Bekijk cursus

Interactieve oefening met praktijkervaring

Probeer deze oefening door deze voorbeeldcode aan te vullen.

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)
Code bewerken en uitvoeren