Skip to content

Instantly share code, notes, and snippets.

@sararob

sararob/multi-label-head.py Secret

Created August 14, 2018 18:25
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save sararob/d9215efe0a6b6cad9dcca9b84b4f9e24 to your computer and use it in GitHub Desktop.
Save sararob/d9215efe0a6b6cad9dcca9b84b4f9e24 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
multi-label-head.py
multi_label_head = tf.contrib.estimator.multi_label_head(
num_classes,
loss_reduction=tf.losses.Reduction.SUM_OVER_BATCH_SIZE
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment