Od regresji do klasyfikacji wieloklasowej
Modele przeznaczone do klasyfikacji binarnej, klasyfikacji wieloklasowej i regresji są do siebie podobne – różnią się jedynie kilkoma szczegółami.
Zacznij od zbudowania modelu do regresji, a następnie zmodyfikuj go tak, by wykonywał klasyfikację wieloklasową.
To ćwiczenie jest częścią kursu
Wprowadzenie do uczenia głębokiego z PyTorch
Interaktywne ćwiczenie praktyczne
Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.
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)