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