Van regressie naar multi-class-classificatie
De modellen die je hebt gezien voor binaire classificatie, multi-class-classificatie en regressie leken allemaal op elkaar, op een paar kleine aanpassingen na.
Begin met het bouwen van een model voor regressie, en pas het model daarna aan om een multi-class-classificatie uit te voeren.
Deze oefening maakt deel uit van de cursus
Introductie tot Deep Learning met PyTorch
Praktische interactieve oefening
Probeer deze oefening eens door deze voorbeeldcode in te vullen.
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)