Ein Vorhersagemodell mit RNN erstellen
Es ist Zeit, dein erstes rekurrentes Netz zu bauen! Es wird ein Sequenz-zu-Vektor-Modell sein, das aus einer RNN-Schicht mit zwei Schichten und einer hidden_size
von 32
besteht. Nach der RNN-Schicht bildet eine einfache lineare Schicht die Outputs auf einen einzelnen vorherzusagenden Wert ab.
Die folgenden Imports wurden bereits für dich durchgeführt:
import torch
import torch.nn as nn
Diese Übung ist Teil des Kurses
Deep Learning mit PyTorch für Fortgeschrittene
Interaktive Übung
Versuche dich an dieser Übung, indem du diesen Beispielcode vervollständigst.
class Net(nn.Module):
def __init__(self):
super().__init__()
# Define RNN layer
self.rnn = ____(
____,
____,
____,
____,
)
self.fc = nn.Linear(32, 1)