加重クロスエントロピーテンソルの流れ

Question

1に足りないラベルを渡すことができるテンソルフローの組み込みが見つかりませんでした。したがって、自分自身で作成しようとしました：（入力は[batch_size、labels]）

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)) 

しかし、それは機能していないようです（損失は発散しています）。私は間違ったことをしましたか？

Answer 1

私はそれが正しいと思いますが、アダムオプティマイザの「イプシロン」を低く設定しました。