Construcción de una previsión RNN
Es hora de construir tu primera red recurrente. Será un modelo de secuencia a vector compuesto por una capa RNN con dos capas y un hidden_size
de 32
. Después de la capa RNN, una capa lineal simple asignará las salidas a un único valor a predecir.
Las siguientes importaciones ya se han realizado por ti:
import torch
import torch.nn as nn
Este ejercicio forma parte del curso
Aprendizaje profundo intermedio con PyTorch
Ejercicio interactivo práctico
Pruebe este ejercicio completando este código de muestra.
class Net(nn.Module):
def __init__(self):
super().__init__()
# Define RNN layer
self.rnn = ____(
____,
____,
____,
____,
)
self.fc = nn.Linear(32, 1)