Construire un classificateur binaire avec PyTorch
Rappelons qu’un petit réseau neuronal comportant une seule couche linéaire suivie d’une fonction sigmoïde est un classificateur binaire. Il se comporte comme une régression logistique.
Exercez-vous à construire ce petit réseau et à interpréter la sortie du classificateur.
Cet exercice fait partie du cours
<cours>Introduction au deep learning avec PyTorch</cours>Exercice interactif pratique
Essayez cet exercice en complétant ce code d’exemple.
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)