Skip to content
This repository has been archived by the owner on Nov 3, 2022. It is now read-only.

Added dice_loss in keras_contrib/losses with docstring and pytest functionality #524

Open
wants to merge 6 commits into
base: master
Choose a base branch
from

Conversation

soymintc
Copy link

- What I did
Added Dice coefficient loss function with documentation in keras_contrib/losses, and modified keras_contrib/losses/init.py accordingly
Added pytest functionality

- How I did it
Added dice_loss function in keras_contrib/losses/dice.py, and verified it using pytest.
Testing script is added in tests/keras_contrib/losses/dice_test.py

- How you can verify it

  1. Ran pytest dice_test.py inside tests/keras_contrib/losses, which all passed
  2. Uninstalled the package and re-installed keras_contrib, and re-did step 1, which had no problem.
  3. Ran pytest in the base directory, which didn't result in errors for tests/keras_contrib/losses/dice_test.py

This pull request fixes #484

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Add Dice coefficient to metrics and Dice coefficient loss to losses
1 participant