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