Da regressão à classificação multiclasse
Os modelos que você viu para classificação binária, classificação multiclasse e regressão foram todos semelhantes, exceto por alguns ajustes no modelo.
Comece a criar um modelo para regressão e, em seguida, ajuste o modelo para realizar uma classificação multiclasse.
Este exercício faz parte do curso
Introdução ao Aprendizado Profundo com o PyTorch
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
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)