search

pytorch / ignite

High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.
https://pytorch-ignite.ai
BSD 3-Clause "New" or "Revised" License
4.51k stars 612 forks source link

Rewrite test for Entropy metric #3216

Closed kzkadc closed 6 months ago

kzkadc commented 6 months ago

Fixes https://github.com/pytorch/ignite/pull/3210#discussion_r1535393084

Description:

Check list:

kzkadc commented 6 months ago

@vfdev-5 I updated the test for Entropy metric, but I'm not certain if it's correct.

vfdev-5 commented 6 months ago

@kzkadc thanks a lot for the draft, I'll push the corrections to the PR directly and we can land it after that