Sestavení predikcí pomocí RNN
Je čas sestavit svou první rekurentní síť! Půjde o model typu sekvence-na-vektor, který se skládá z vrstvy RNN se dvěma vrstvami a hidden_size o hodnotě 32. Za vrstvou RNN bude jednoduchá lineární vrstva mapující výstupy na jedinou predikovanou hodnotu.
Následující importy už máš připravené:
import torch
import torch.nn as nn
Toto cvičení je součástí kurzu
Intermediate Deep Learning with PyTorch
Interaktivní cvičení na vyzkoušení si v praxi
Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.
class Net(nn.Module):
def __init__(self):
super().__init__()
# Define RNN layer
self.rnn = ____(
____,
____,
____,
____,
)
self.fc = nn.Linear(32, 1)