Limit the plots displayed through `.show()` function

[lsickert]

Limit the plots displayed through .show() function

Description

Currently, the .show() function on the attribution output will display plots for all generated attributions by default. For larger batches, this can lead to huge outputs in a notebook/ large numbers of HTML files being generated. It might be preferable to provide a sensible default value to the number of plots that are displayed in a notebook and allow users to specify for themselves how many/which attributions they want to have visualized by referring to their index.

Similar functionality is already possible now by manually choosing the indices of out.sequence_attributions and calling the .show() methods of the individual attribution outputs, so this function would mainly be a convenience function for new users.

Motivation

see above

Additional context

Commit to Help

• [x] I'm willing to help with this feature.