1

私はテンソルフローを使用し、ニューラルネットワークを使用してマルチラベル分類の問題を解決しています。損失関数として Softmax クロス エントロピーを使用していました。

#Softmax loss
cost = tf.reduce_mean(tf.nn.softmax_cross_entropy_with_logits(pred, y))

さて、KL発散損失関数を使うべきだと思いました。しかし、私はテンソルフローでそれを見つけられませんでした.Softmax損失の代わりにKL Divergence損失関数を使用するのに役立つ体はありますか?

4

1 に答える 1

0

どうぞ：

tf.contrib.distributions.kl(distribution_1, distribution_2)
于 2016-10-21T12:39:36.577 に答える