Keras: classe de poids (class_weight) pour un chaud encodage

Je voudrais utiliser class_weight argument en keras modèle.aptes à gérer le déséquilibre des données d'entraînement. En regardant certains documents, j'ai compris que nous pouvons passer d'un dictionnaire comme ceci:

class_weight = {0 : 1,
    1: 1,
    2: 5}

(Dans cet exemple, la classe 2 est plus élevé peine de la perte de fonction.)

Le problème est que mon réseau de sortie a un chaud encodage c'est à dire la classe-0 = (1, 0, 0), classe-1 = (0, 1, 0), et la classe-3 = (0, 0, 1).

Comment pouvons-nous utiliser la class_weight pour une chaude de sortie encodé?

En regardant certains codes dans Keras, il ressemble à _feed_output_names contiennent une liste de sortie des classes, mais dans mon cas, model.output_names/model._feed_output_names retourne ['dense_1']

Connexes: Comment définir la classe de poids pour le déséquilibre des classes dans Keras?

OriginalL'auteur Naoto Usuyama | 2017-04-18