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 el código de muestra.
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)