CommencerCommencer gratuitement

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

Introduction au deep learning avec PyTorch

Afficher le cours

Exercice interactif pratique

Essayez cet exercice en complétant cet exemple de code.

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)
Modifier et exécuter le code