จาก Regression สู่ Multi-class Classification
โมเดลสำหรับ binary classification, multi-class classification และ regression ที่ได้เรียนมานั้นมีโครงสร้างคล้ายกัน โดยต่างกันเพียงการปรับแต่งเล็กน้อย
เริ่มต้นด้วยการสร้างโมเดลสำหรับ regression แล้วปรับโมเดลให้ทำงานแบบ multi-class classification
แบบฝึกหัดนี้เป็นส่วนหนึ่งของหลักสูตร
Deep Learning เบื้องต้นด้วย PyTorch
แบบฝึกหัดเชิงโต้ตอบแบบลงมือทำ
ลองทำแบบฝึกหัดนี้โดยเติมโค้ดตัวอย่างนี้ให้สมบูรณ์
import torch
import torch.nn as nn
input_tensor = torch.Tensor([[3, 4, 6, 7, 10, 12, 2, 3, 6, 8, 9]])
# Implement a neural network with exactly four linear layers
model = nn.____(
____
)
output = model(input_tensor)
print(output)