Improve get_indices_for_active_step by caching the result according t…

baal-org / baal

Bayesian active learning library for research and industrial usecases.

https://baal.readthedocs.io

Apache License 2.0

854 stars 84 forks source link

Improve get_indices_for_active_step by caching the result according t… #265

Closed Dref360 closed 1 year ago

Dref360 commented 1 year ago

…o the current AL step

Summary:

A faster way would be to clear the cache only when we call label, but that would break the encapsulation IMO. So leaving it at that, it is still 10x better.

Features:

Checklist:

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