search

g8a9 / ferret

A python package for benchmarking interpretability techniques on Transformers.
https://ferret.readthedocs.io
MIT License
209 stars 16 forks source link

Prediction in batches #4

Closed elianap closed 2 years ago

elianap commented 2 years ago

Add prediction in batches for LOO&faithfulness evaluations (model wrapper: _get_class_predicted_probabilities_texts)

g8a9 commented 2 years ago

Implemented in https://github.com/g8a9/ferret/releases/tag/0.4.0 !