ComenzarEmpieza gratis

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

Ver curso

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)
Editar y ejecutar código