Creación de una RNN de predicción
Es hora de construir tu primera red recurrente. Será un modelo secuencia-vector compuesto por una capa RNN con dos capas y un e hidden_size e de 32. Después de la capa RNN, una capa lineal simple asignará las salidas a un único valor que se va 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
Prueba este ejercicio y completa el código de muestra.
class Net(nn.Module):
def __init__(self):
super().__init__()
# Define RNN layer
self.rnn = ____(
____,
____,
____,
____,
)
self.fc = nn.Linear(32, 1)