重み付きloss

loss.py

def loss(labels, logits):


    weights = tf.where(labels == 0, 1, 1)
    weights = tf.where(labels == 1, 1, weights)
    weights = tf.where(labels == 2, 2, weights)
    weights = tf.where(labels == 3, 20, weights)
    weights = tf.where(labels == 4, 20, weights)
    weights = tf.where(labels == 5, 20, weights)

    classify_loss = tf.losses.sparse_softmax_cross_entropy(labels=labels, logits=logits, scope='loss', weights=weights)
    return classify_loss