search

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

Handle NER usecase #263

Closed Dref360 closed 11 months ago

Dref360 commented 1 year ago

Slight modification to handle NER.

NER pipeline outputs [?, Sequence, NClass], but we expect [?, NClass, Sequence]. So I changed that in predict_on_dataset_generator.