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