1. Học hỏi
    2. /
  2. Khoa Học
    3. /
  3. Deep Learning cho Ảnh với PyTorch
Connected

Bài tập

Mô hình phân loại đa lớp

Với mẫu cho mô hình phân loại nhị phân đã có sẵn, giờ bạn có thể dựa vào đó để thiết kế mô hình phân loại đa lớp. Mô hình cần xử lý số lượng lớp khác nhau thông qua một tham số, cho phép bạn tùy chỉnh mô hình cho các bài toán phân loại đa lớp cụ thể trong tương lai.

Các gói torch và torch.nn với bí danh nn đã được nhập. Tất cả ảnh đều có kích thước 64x64 pixel.

Hướng dẫn

100 XP
  • Định nghĩa phương thức __init__ bao gồm các tham số self và num_classes.
  • Tạo một tầng fully connected với kích thước đầu vào 16*32*32 và số lớp num_classes là đầu ra.
  • Tạo hàm kích hoạt softmax với dim=1.