Add parameter axis to tversky loss

keras-team / keras

Deep Learning for humans

http://keras.io/

Apache License 2.0

62.13k stars 19.49k forks source link

Open jakubxy08 opened 1 week ago

jakubxy08 commented 1 week ago

Add parameter axis to tversky loss similar to dice loss. I can implement that if someone give me green light :)

sachinprasadhs commented 1 week ago

Feel free to create a PR, Keras team would be happy to review. Thanks