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

ประเมินความแม่นยำของโมเดลด้วย Torchmetrics

การประเมินประสิทธิภาพของโมเดลเป็นสิ่งสำคัญ โดยเฉพาะเมื่อเตรียมนำไปใช้งานจริง มาผสานการคำนวณค่าความแม่นยำด้วย Torchmetrics เข้าไปใน validation_step() กัน อย่าลืม log ผลลัพธ์ด้วย เพื่อให้ติดตามความคืบหน้าของโมเดลได้อย่างสะดวก

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

AI Models แบบ Scalable ด้วย PyTorch Lightning

ดูคอร์ส

คำแนะนำการฝึกหัด

  • Import Accuracy จาก torchmetrics
  • สร้าง accuracy metric ภายใน __init__()
  • คำนวณค่าความแม่นยำใน validation_step() และ log ผลลัพธ์ในชื่อ 'val_acc'

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

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

# Import relevant metric
from torchmetrics import ____
import lightning.pytorch as pl

class ClassifierModel(pl.LightningModule):
    def __init__(self):
        super().__init__()
        # Instantiate accuracy metric
        self.accuracy = ____()
    def validation_step(self, batch, batch_idx):
        x, y = batch
        preds = self(x)
        # Calculate accuracy and log it as val_acc
        acc = self.____(preds, y)
        self.log(____, acc)
แก้ไขและรันโค้ด