CommencerCommencez 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

<cours>Deep learning intermédiaire avec PyTorch</cours>
Voir le cours

Exercice interactif pratique

Essayez cet exercice en complétant ce code d’exemple.

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