Bygg en binär klassificerare i PyTorch
Kom ihåg att ett litet neuralt nätverk med ett enda linjärt lager följt av en sigmoidfunktion är en binär klassificerare. Det fungerar precis som en logistisk regression.
Öva på att bygga det här lilla nätverket och tolka klassificerarens utdata.
Den här övningen är en del av kursen
Introduktion till djupinlärning med PyTorch
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
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)