De la regresie la clasificare multi-clasă
Modelele pe care le-ai văzut pentru clasificarea binară, clasificarea multi-clasă și regresie au fost similare, cu câteva ajustări minore.
Începe prin a construi un model de regresie, apoi modifică-l pentru a realiza o clasificare multi-clasă.
Acest exercițiu face parte din cursul
Introducere în Deep Learning cu PyTorch
Exercițiu interactiv practic
Încearcă acest exercițiu completând acest cod de exemplu.
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)