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
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)