1. Nauka
    2. /
  2. Kursy
    3. /
  3. Głębokie uczenie z PyTorch – poziom średnio zaawansowany
Connected

ćwiczenie

Budowanie prognozującej sieci RNN

Czas zbudować swoją pierwszą sieć rekurencyjną! Będzie to model typu sekwencja-do-wektora, składający się z warstwy RNN z dwiema warstwami i hidden_size równym 32. Po warstwie RNN prosta warstwa liniowa przekształci wyjścia w pojedynczą prognozowaną wartość.

Następujące importy zostały już wykonane:

import torch
import torch.nn as nn

Instrukcje 1/4

undefined XP
    1
    2
    3
    4
  • Zdefiniuj warstwę RNN, przekazując jej odpowiednie wartości dla input_size, hidden_size, num_layers i batch_first, a następnie przypisz ją do self.rnn.