Crear un clasificador binario en PyTorch
Recordemos que una pequeña red neuronal con una sola capa lineal seguida de una función sigmoide es un clasificador binario. Actúa igual que una regresión logística.
Practica la creación de esta pequeña red y la interpretación de la salida del clasificador.
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, 2, 3, 6, 8, 9]])
# Implement a small neural network for binary classification
model = nn.Sequential(
nn.____(____),
nn.____()
)
output = model(input_tensor)
print(output)