Kom igångKom igång gratis

One-hot-kodade boktitlar

PyBooks vill katalogisera och analysera bokgenrer i sitt bibliotek. Tillämpa one-hot-kodning på en lista med bokgenrer för att göra dem maskinläsbara.

torch har redan importerats åt dig.

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

Djupinlärning för text med PyTorch

Visa kurs

Övningsinstruktioner

  • Definiera vokabulärens storlek och spara den i vocab_size.
  • Skapa one-hot-vektorer med lämplig metod i torch och vocab_size.
  • Skapa ett lexikon som mappar genrer till deras motsvarande one-hot-vektorer med hjälp av dictionary comprehension; lexikonets nycklar ska vara genren.

Interaktiv övning med praktiskt arbete

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

genres = ['Fiction','Non-fiction','Biography', 'Children','Mystery']

# Define the size of the vocabulary
vocab_size = ____(____)

# Create one-hot vectors
one_hot_vectors = torch.____(____)

# Create a dictionary mapping genres to their one-hot vectors
one_hot_dict = {____: ____[i] for i, genre in enumerate(genres)}

for genre, vector in one_hot_dict.items():
    print(f'{genre}: {vector.numpy()}')
Redigera och kör kod