การสร้าง Binary Classifier ด้วย PyTorch
โครงข่ายประสาทเทียมขนาดเล็กที่มี linear layer เพียงชั้นเดียวตามด้วยฟังก์ชัน sigmoid จะทำหน้าที่เป็น binary classifier ซึ่งทำงานเหมือน logistic regression
ลองฝึกสร้างโครงข่ายขนาดเล็กนี้ และฝึกตีความผลลัพธ์ที่ได้จาก classifier
แบบฝึกหัดนี้เป็นส่วนหนึ่งของหลักสูตร
Deep Learning เบื้องต้นด้วย PyTorch
แบบฝึกหัดเชิงโต้ตอบแบบลงมือทำ
ลองทำแบบฝึกหัดนี้โดยเติมโค้ดตัวอย่างนี้ให้สมบูรณ์
import torch
import torch.nn as nn
input_tensor = torch.Tensor([[3, 4, 6, 2, 3, 6, 8, 9]])
# Implement a small neural network for binary classification
model = nn.Sequential(
nn.____(____),
nn.____()
)
output = model(input_tensor)
print(output)