Kom igångKom igång gratis

Bygg ett prognostiserande RNN

Nu är det dags att bygga ditt första rekurrenta nätverk! Det kommer att vara en sekvens-till-vektor-modell bestående av ett RNN-lager med två lager och en hidden_size på 32. Efter RNN-lagret mappar ett enkelt linjärt lager utdata till ett enda värde som ska predikteras.

Följande importer har redan gjorts åt dig:

import torch
import torch.nn as nn

Den här övningen är en del av kursen

Fördjupad djupinlärning med PyTorch

Visa kurs

Interaktiv övning med praktiskt arbete

Testa den här övningen genom att slutföra den här exempelkoden.

class Net(nn.Module):
    def __init__(self):
        super().__init__()
        # Define RNN layer
        self.rnn = ____(
            ____,
            ____,
            ____,
            ____,
        )
        self.fc = nn.Linear(32, 1)
Redigera och kör kod