От регрессии к многоклассовой классификации
Модели для бинарной классификации, многоклассовой классификации и регрессии очень похожи — между ними лишь небольшие различия.
Для начала постройте модель для регрессии, а затем адаптируйте её для многоклассовой классификации.
Это упражнение является частью курса
Введение в глубокое обучение с PyTorch
Интерактивное практическое упражнение
Попробуйте выполнить это упражнение, дополнив этот пример кода.
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)