CommencezCommencez gratuitement

De la régression à la classification multiclasse

Les modèles que vous avez vus pour la classification binaire, la classification multiclasse et la régression se ressemblent tous, à quelques ajustements près.

Commencez par bâtir un modèle de régression, puis modifiez-le pour effectuer une classification multiclasse.

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, 7, 10, 12, 2, 3, 6, 8, 9]])

# Implement a neural network with exactly four linear layers
model = nn.____(
  ____
)

output = model(input_tensor)
print(output)
Modifier et exécuter le code