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 y completa 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)