1. Learn
  2. /
  3. Cursuri
  4. /
  5. Deep Learning intermediar cu PyTorch

Connected

exercițiu

Construirea unui RNN de prognoză

E momentul să construiești prima ta rețea recurentă! Va fi un model de tip secvență-la-vector, alcătuit dintr-un strat RNN cu două niveluri și un hidden_size de 32. După stratul RNN, un strat liniar simplu va transforma rezultatele într-o singură valoare de prezis.

Următoarele importuri au fost deja realizate pentru tine:

import torch
import torch.nn as nn

Instrucțiuni 1/4

undefined XP
    1
    2
    3
    4
  • Definește stratul RNN transmițându-i valorile corecte pentru input_size, hidden_size, num_layers și batch_first, apoi atribuie-l lui self.rnn.