IniziaInizia 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 il corso

Esercizio pratico interattivo

Prova a risolvere questo esercizio completando il 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