Từ hồi quy sang phân loại đa lớp
Các mô hình bạn đã thấy cho bài toán phân loại nhị phân, phân loại đa lớp và hồi quy đều khá giống nhau, chỉ khác một vài điều chỉnh nhỏ.
Hãy bắt đầu xây dựng một mô hình cho hồi quy, sau đó tinh chỉnh mô hình để thực hiện phân loại đa lớp.
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, 7, 10, 12, 2, 3, 6, 8, 9]])
# Implement a neural network with exactly four linear layers
model = nn.____(
____
)
output = model(input_tensor)
print(output)