CommencezCommencez gratuitement

Titres de livres en one-hot encoding

PyBooks souhaite cataloguer et analyser les genres de livres de sa bibliothèque. Appliquez un one-hot encoding à une liste de genres afin de les rendre lisibles par machine.

torch a été importé pour vous.

Cette activité fait partie du cours

Apprentissage profond pour le texte avec PyTorch

Voir le cours

Instructions de l’exercice

  • Définissez la taille du vocabulaire et enregistrez-la dans vocab_size.
  • Créez des vecteurs one-hot en utilisant la technique appropriée de torch et vocab_size.
  • Créez un dictionnaire qui associe chaque genre à son vecteur one-hot correspondant au moyen d'une compréhension de dictionnaire; les clés du dictionnaire doivent être le genre.

Exercice interactif pratique

Essayez cet exercice en complétant ce code d’exemple.

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()}')
Modifier et exécuter le code