1. Apprendre
    2. /
  2. Cours
    3. /
  3. Modélisation d'images avec Keras
Connected

Exercice

Représenter des images avec l'encodage one-hot

Les réseaux de neurones s'attendent à ce que les étiquettes de classes d'un jeu de données soient organisées au format one-hot : chaque ligne du tableau contient des zéros dans toutes les colonnes, sauf dans la colonne correspondant à une étiquette donnée, qui est mise à 1.

Le jeu de données sur la mode contient trois catégories :

  1. Chemises
  2. Robes
  3. Chaussures

Dans cet exercice, vous allez créer un encodage one-hot pour un petit échantillon de ces étiquettes.

Instructions

100 XP
  • Initialisez la variable ohe_labels pour contenir le tableau encodé en one-hot.
  • Utilisez np.where() pour trouver l'emplacement de la catégorie de l'article à chaque itération dans categories.
  • Inscrivez un 1 à la bonne combinaison ligne/colonne à chaque itération.