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

Bài tập

Xây dựng bộ phân loại nhị phân trong PyTorch

Hãy nhớ rằng một mạng nơ-ron nhỏ với một lớp tuyến tính duy nhất tiếp theo là hàm sigmoid chính là bộ phân loại nhị phân. Nó hoạt động giống hệt hồi quy logistic.

Hãy luyện tập xây dựng mạng nhỏ này và diễn giải đầu ra của bộ phân loại.

Hướng dẫn 1/2

undefined XP
    1
    2
  • Tạo một mạng nơ-ron nhận tensor 1x8 làm đầu vào và xuất ra một giá trị đơn cho bài toán phân loại nhị phân.
  • Truyền đầu ra của lớp tuyến tính qua sigmoid để tạo ra một xác suất.