Une première façon de trouver la matrice d'embedding est de l’entraîner sur un problème supervisé. De ce fait, nous allons utiliser cette représentation des mots pour résoudre un problème de sentiment analysis sur les reviews IMBD. Le jeu de données comporte 25000 critiques de films (plus d'informations sur le jeu de données choisi). Il est alors possible d'entraîner la matrice W du word embedding en même temps que l'entraînement du problème de classification.