EmpezarEmpieza 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 completando este código de ejemplo.

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