ComenzarEmpieza gratis

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

Ver curso

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)
Editar y ejecutar código