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
Vervollständige den Beispielcode, um diese Übung erfolgreich abzuschließen.
class Net(nn.Module):
def __init__(self):
super().__init__()
# Define RNN layer
self.rnn = ____(
____,
____,
____,
____,
)
self.fc = nn.Linear(32, 1)