Skip to content

Instantly share code, notes, and snippets.

@eustin

eustin/letor-part-2-kl-divergence-tf.py

Created June 8, 2020 04:35
Show Gist options
  • Save eustin/634a2a3d5cee68f425946518af4301fd to your computer and use it in GitHub Desktop.
Save eustin/634a2a3d5cee68f425946518af4301fd to your computer and use it in GitHub Desktop.
Download ZIP
Raw
letor-part-2-kl-divergence-tf.py
loss = tf.keras.losses.KLDivergence()
batch_loss = loss(relevance_grades_prob_dist, scores_prob_dist)
print(batch_loss)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment