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
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)