Inizia subitoInizia gratis

Creare un classificatore binario in PyTorch

Ricorda che una piccola rete neurale con un singolo livello lineare seguito da una funzione sigmoid è un classificatore binario. Si comporta proprio come una regressione logistica.

Esercitati a costruire questa piccola rete e a interpretare l'output del classificatore.

Questo esercizio fa parte del corso

Introduzione al Deep Learning con PyTorch

Visualizza corso

esercizio interattivo pratico

Prova questo esercizio completando questo codice di esempio.

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)
Modifica ed esegui il codice