Aan de slagGa gratis aan de slag

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

Cursus bekijken

Praktische interactieve oefening

Probeer deze oefening eens door deze voorbeeldcode in 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