Від регресії до багатокласової класифікації
Моделі, які ви бачили для бінарної класифікації, багатокласової класифікації та регресії, дуже схожі, за винятком кількох налаштувань у самій моделі.
Почніть будувати модель для регресії, а потім змініть її, щоб виконувати багатокласову класифікацію.
Ця вправа є частиною курсу
Вступ до Deep Learning з 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)