Od regrese k vícetřídní klasifikaci
Modely pro binární klasifikaci, víceřídní klasifikaci a regresi jsou si navzájem podobné – liší se jen v několika drobných úpravách.
Začni postavením modelu pro regresi a pak ho uprav tak, aby zvládl víceřídní klasifikaci.
Toto cvičení je součástí kurzu
Introduction to Deep Learning with PyTorch
Interaktivní cvičení na vyzkoušení si v praxi
Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.
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)