Från regression till flerklass-klassificering
De modeller du har sett för binär klassificering, flerklass-klassificering och regression har alla varit liknande, med några få justeringar.
Börja med att bygga en modell för regression, och anpassa den sedan för att utföra en flerklass-klassificering.
Den här övningen är en del av kursen
Introduktion till djupinlärning med PyTorch
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
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)