CommencerCommencer 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

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