Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

New cross-entropy losses with temperature #77

Merged
merged 4 commits into from
Oct 10, 2023

Conversation

cofri
Copy link
Collaborator

@cofri cofri commented Jul 28, 2023

This PR introduces two new cross-entropy losses based on Keras standard losses and with a settable temperature for softmax:

  • TauSparseCategoricalCrossentropy equivalent to Keras SparseCategoricalCrossentropy
  • TauBinaryCrossentropy equivalent to Keras BinaryCrossentropy

Copy link
Collaborator

@franckma31 franckma31 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

looks fine for me

@cofri cofri marked this pull request as ready for review August 21, 2023 14:35
@cofri cofri requested a review from thib-s August 21, 2023 14:36
A new loss is introduced: the SparseCategoricalCrossentropy with
the temperature parameter.
A new loss is introduced: the equivalent to BinaryCrossentropy with
a temperature parameter.
@cofri cofri force-pushed the feat/TauSparseCategoricalCrossentropy branch from ef19e3e to f2c5da2 Compare August 21, 2023 14:38
Copy link
Member

@thib-s thib-s left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

looks good to me

@cofri cofri merged commit 47a4c3f into master Oct 10, 2023
5 checks passed
@cofri cofri deleted the feat/TauSparseCategoricalCrossentropy branch October 10, 2023 08:42
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants