tf.nn.softmax_cross_entropy_with_logits()関数は、クロスエントロピーTensorFlowコモンを算出する機能です。
それ以降のバージョン、TensorFlow更新:tf.nn.softmax_cross_entropy_with_logits_v2
形式は次のとおりです。
cross_entropy = tf.nn.softmax_cross_entropy_with_logits(labels=y_, logits=y )
パラメータ:
ラベル:分類ラベルの
logits:予測値
TensorFlowコード実装のtf.nn.softmax_cross_entropy_with_logits()関数:
import tensorflow as tf
labels = [[0.1, 0.2, 0.6],
[0.4, 0.9, 0.7]]
logits = [[1, 2, 4],
[0.1, -1, 3]]
# 计算交叉熵(tf.nn.softmax_cross_entropy_with_logits更新为格
# tf.nn.softmax_cross_entropy_with_logits_v2)
cross_entropy = tf.nn.softmax_cross_entropy_with_logits_v2(labels=labels, logits=logits)
with tf.Session() as sess:
print(sess.run(cross_entropy))
結果:
[0.85286146 4.9015484 ]