CommencezCommencez gratuitement

Créer un classifieur binaire avec PyTorch

Rappelez-vous qu'un petit réseau de neurones composé d'une seule couche linéaire suivie d'une fonction sigmoïde est un classifieur binaire. Il se comporte comme une régression logistique.

Exercez-vous à construire ce petit réseau et à interpréter la sortie du classifieur.

Cette activité fait partie du cours

Introduction au Deep Learning avec PyTorch

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