Tensorflow 가중 교차 엔트로피는 사용이 간단합니다.

logits=[[0.2,0.3,0.4,0.1]]
one_hot_labels = tf.one_hot([2],depth=4,dtype=tf.float32)
loss_ = tf.nn.weighted_cross_entropy_with_logits(labels=one_hot_labels, logits=logits, pos_weight=0.1)
with tf.Session() as session:
    print(session.run(tf.reduce_mean(loss_)))

 

추천

출처blog.csdn.net/u011939633/article/details/103499750