Fix indice remapping to get the correct label

baal-org / baal

Bayesian active learning library for research and industrial usecases.

https://baal.readthedocs.io

Apache License 2.0

862 stars 86 forks source link

Closed Dref360 closed 2 years ago

Dref360 commented 2 years ago

Small issue in this notebook. We wouldn't do the remapping correctly to get the labels.

Fixes #200

[x] Your code is documented (To validate this, add your module to tests/documentation_test.py).
[x] Your code is tested with unit tests.
[x] You moved your Issue to the PR state.