ÎncepețiÎncepe gratuit

Construirea unui clasificator binar în PyTorch

Reține că o rețea neuronală simplă, cu un singur strat liniar urmat de o funcție sigmoid, este un clasificator binar – funcționează exact ca o regresie logistică.

Exersează construirea acestei rețele mici și interpretarea rezultatelor clasificatorului.

Acest exercițiu face parte din cursul

Introducere în Deep Learning cu PyTorch

Vezi cursul

Exercițiu interactiv practic

Încearcă acest exercițiu completând acest cod de exemplu.

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)
Editează și rulează codul