1. Học hỏi
  2. /
  3. Khoa Học
  4. /
  5. Nhập môn Deep Learning với PyTorch

Connected

Bài tập

Tính độ chính xác với torchmetrics

Theo dõi độ chính xác trong quá trình huấn luyện giúp bạn xác định epoch có hiệu suất tốt nhất.

Trong bài tập này, bạn sẽ dùng torchmetrics để tính độ chính xác trên bộ dữ liệu facemask gồm ba lớp. Hàm plot_errors sẽ làm nổi bật các mẫu bị phân loại sai, giúp bạn phân tích lỗi của mô hình.

Gói torchmetrics đã được import sẵn. outputs của mô hình là xác suất softmax, và labels là các vector one-hot.

Hướng dẫn 1/2

undefined XP
    1
    2
  • Tạo metric độ chính xác "multiclass" với ba lớp.
  • Tính độ chính xác cho mỗi batch của dataloader.