CommencerCommencez gratuitement

De la régression à la classification multiclasses

Les modèles que vous avez vus pour la classification binaire, la classification multi-classes et la régression sont tous similaires, à l’exception de quelques ajustements du modèle.

Commencez par construire un modèle de régression, puis modifiez-le pour effectuer une classification multi-classes.

Cet exercice fait partie du cours

<cours>Introduction au deep learning avec PyTorch</cours>
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