Criação de um RNN de previsão
Chegou a hora de você criar sua primeira rede recorrente! Será um modelo de sequência para vetor que consiste em uma camada RNN com duas camadas e uma hidden_size
de 32
. Após a camada RNN, uma camada linear simples mapeará as saídas para um único valor a ser previsto.
As seguintes importações já foram feitas para você:
import torch
import torch.nn as nn
Este exercício faz parte do curso
Aprendizagem profunda intermediária com PyTorch
Exercício interativo prático
Experimente este exercício preenchendo este código de exemplo.
class Net(nn.Module):
def __init__(self):
super().__init__()
# Define RNN layer
self.rnn = ____(
____,
____,
____,
____,
)
self.fc = nn.Linear(32, 1)