เริ่มต้นใช้งานเริ่มต้นใช้งานได้ฟรี

การคำนวณความแม่นยำด้วย torchmetrics

การติดตามความแม่นยำระหว่างการเทรนช่วยระบุได้ว่า epoch ใดให้ผลลัพธ์ดีที่สุด

ในแบบฝึกหัดนี้ คุณจะใช้ torchmetrics เพื่อคำนวณความแม่นยำบน ชุดข้อมูลหน้ากากอนามัย ที่มีสามคลาส ฟังก์ชัน plot_errors จะช่วยแสดงตัวอย่างที่โมเดลจัดประเภทผิด เพื่อให้วิเคราะห์ข้อผิดพลาดได้ง่ายขึ้น

แพ็กเกจ torchmetrics ถูก import ไว้แล้ว outputs ของโมเดลเป็นความน่าจะเป็นแบบ softmax และ labels เป็นเวกเตอร์แบบ one-hot encoded

แบบฝึกหัดนี้เป็นส่วนหนึ่งของหลักสูตร

Deep Learning เบื้องต้นด้วย PyTorch

ดูคอร์ส

แบบฝึกหัดเชิงโต้ตอบแบบลงมือทำ

ลองทำแบบฝึกหัดนี้โดยเติมโค้ดตัวอย่างนี้ให้สมบูรณ์

# Create accuracy metric
metric = torchmetrics.____(____, ____)
for features, labels in dataloader:
    outputs = model(features)
  
    # Calculate accuracy over the batch
    metric.____(____, ____)
แก้ไขและรันโค้ด