CommencerCommencer gratuitement

Construire un RNN de prévision

Il est temps de construire votre premier réseau récurrent ! Il s’agira d’un modèle séquence-vers-vecteur composé d’une couche RNN avec deux couches et un hidden_size de 32. Après la couche RNN, une simple couche linéaire projettera les sorties vers une valeur unique à prédire.

Les imports suivants ont déjà été effectués pour vous :

import torch
import torch.nn as nn

Cet exercice fait partie du cours

Deep learning intermédiaire avec PyTorch

Afficher le cours

Exercice interactif pratique

Essayez cet exercice en complétant cet exemple de code.

class Net(nn.Module):
    def __init__(self):
        super().__init__()
        # Define RNN layer
        self.rnn = ____(
            ____,
            ____,
            ____,
            ____,
        )
        self.fc = nn.Linear(32, 1)
Modifier et exécuter le code