สร้าง RNN สำหรับการพยากรณ์
ได้เวลาสร้าง recurrent network ตัวแรกแล้ว! โมเดลนี้จะเป็นแบบ sequence-to-vector ประกอบด้วย RNN layer ที่มี 2 layers และ hidden_size เท่ากับ 32 จากนั้น linear layer จะแมปผลลัพธ์ให้เหลือค่าเดียวสำหรับการพยากรณ์
ได้ import ไลบรารีต่อไปนี้ไว้ให้แล้ว:
import torch
import torch.nn as nn
แบบฝึกหัดนี้เป็นส่วนหนึ่งของหลักสูตร
Deep Learning ระดับกลางด้วย PyTorch
แบบฝึกหัดเชิงโต้ตอบแบบลงมือทำ
ลองทำแบบฝึกหัดนี้โดยเติมโค้ดตัวอย่างนี้ให้สมบูรณ์
class Net(nn.Module):
def __init__(self):
super().__init__()
# Define RNN layer
self.rnn = ____(
____,
____,
____,
____,
)
self.fc = nn.Linear(32, 1)