Xây dựng một RNN dự báo
Đến lúc xây dựng mạng hồi quy đầu tiên của bạn! Đây sẽ là một mô hình sequence-to-vector gồm một tầng RNN với hai lớp và hidden_size là 32. Sau tầng RNN, một tầng tuyến tính đơn giản sẽ ánh xạ đầu ra thành một giá trị duy nhất cần dự đoán.
Các import sau đã được chuẩn bị sẵn cho bạn:
import torch
import torch.nn as nn
Bài tập này là một phần của khóa học
Deep Learning nâng cao với PyTorch
Bài tập tương tác thực hành trực tiếp
Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.
class Net(nn.Module):
def __init__(self):
super().__init__()
# Define RNN layer
self.rnn = ____(
____,
____,
____,
____,
)
self.fc = nn.Linear(32, 1)