Gewichtete Kreuz-Entropie Tensorflow

Question

Ich konnte nicht eine Tensorflow-integrierte finden, die Sie in Etiketten, die Summe nicht 1, übergeben können, so versucht, meine eigene schreiben: (Eingabe ist [batch_size, Etiketten])

tf.reduce_mean(tf.reduce_sum(y_true,axis=1) * tf.reduce_logsumexp(y_pred_logits,axis=1) 
- tf.reduce_sum(y_true * y_pred_logits,axis=1)) 

Es scheint jedoch nicht zu funktionieren (Verlust divergiert). Habe ich etwas falsch gemacht?

Answer 1

Ich denke, es war richtig, aber ich habe das "Epsilon" von Adam Optimizer zu niedrig gesetzt.