De la regresión a la clasificación multiclase
Los modelos que has visto para la clasificación binaria, la clasificación multiclase y la regresión han sido todos similares, salvo algunos ajustes del modelo.
Empieza construyendo un modelo para la regresión y luego ajústalo para realizar una clasificación multiclase.
Este ejercicio forma parte del curso
Introducción al aprendizaje profundo con PyTorch
ejercicio interactivo práctico
Prueba este ejercicio completando este código de ejemplo.
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)