1. Učit se
    2. /
  2. Kurzy
    3. /
  3. Intermediate Deep Learning with PyTorch
Connected

cvičení

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

Pokyny 1/4

undefined XP
    1
    2
    3
    4
  • Definuj vrstvu RNN se správnými hodnotami pro input_size, hidden_size, num_layers a batch_first a přiřaď ji do self.rnn.