CommencerCommencer gratuitement

Construire un système de prévision RNN

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

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

import torch
import torch.nn as nn

Cet exercice fait partie du cours

Apprentissage profond 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