Kom igångKom igång gratis

Skapa one-hot-kodade etiketter

One-hot-kodning omvandlar en heltalssetikett till en vektor med N element, där N är antalet klasser. Vektorn innehåller nollor och en etta på rätt position.

I den här övningen skapar du manuellt en one-hot-kodad vektor för y, och använder sedan PyTorch för att förenkla processen. Din datamängd har tre klasser (0, 1, 2).

numpy (np), torch.nn.functional (F) och torch är redan importerade.

Den här övningen är en del av kursen

Introduktion till djupinlärning med PyTorch

Visa kurs

Övningsinstruktioner

  • Utför one-hot-kodning manuellt för det sanna etikettsvärdet y med hjälp av den tillhandahållna NumPy-arrayen och spara resultatet som one_hot_numpy.
  • Använd PyTorch för att one-hot-koda y och spara resultatet som one_hot_pytorch.

Interaktiv övning med praktiskt arbete

Testa den här övningen genom att slutföra den här exempelkoden.

y = 1
num_classes = 3

# Create the one-hot encoded vector using NumPy
one_hot_numpy = np.array([____, ____, ____])

# Create the one-hot encoded vector using PyTorch
one_hot_pytorch = F.____(torch.tensor(y), num_classes=____)

print("One-hot vector using NumPy:", one_hot_numpy)
print("One-hot vector using PyTorch:", one_hot_pytorch)
Redigera och kör kod