CommencezCommencez gratuitement

Créer un RNN de prévision

Il est temps de bâtir votre premier réseau récurrent ! Ce sera 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 fera la correspondance des sorties vers une valeur unique à prédire.

Les importations suivantes ont déjà été faites pour vous :

import torch
import torch.nn as nn

Cette activité fait partie du cours

Apprentissage profond intermédiaire avec PyTorch

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