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.
Bài tập này là một phần của khóa học
Nhập môn Deep Learning với PyTorch
Bài tập tương tác thực hành trực tiếp
Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.
import torch
import torch.nn as nn
input_tensor = torch.Tensor([[3, 4, 6, 2, 3, 6, 8, 9]])
# Implement a small neural network for binary classification
model = nn.Sequential(
nn.____(____),
nn.____()
)
output = model(input_tensor)
print(output)