Budowanie klasyfikatora binarnego w PyTorch
Przypomnij sobie, że mała sieć neuronowa z jedną warstwą liniową, po której następuje funkcja sigmoid, to klasyfikator binarny – działa tak samo jak regresja logistyczna.
Poćwicz budowanie takiej małej sieci i interpretowanie wyników klasyfikatora.
To ćwiczenie jest częścią kursu
Wprowadzenie do uczenia głębokiego z PyTorch
Interaktywne ćwiczenie praktyczne
Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.
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)