ComeçarComece de graça

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

Ver curso

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)
Editar e executar o código