Como criar um classificador binário no PyTorch
Lembre-se de que uma pequena rede neural com uma única camada linear seguida de uma função sigmoide é um classificador binário. Ele funciona exatamente como uma regressão logística.
Pratique a criação dessa pequena rede e interprete a saída do classificador.
Este exercício faz parte do curso
Introdução ao Aprendizado Profundo com o PyTorch
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
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)