search

RasaHQ / rasalit

Visualizations and helpers to improve and debug machine learning models for Rasa Open Source
Apache License 2.0
305 stars 62 forks source link

Displaying attention heatmaps with respect to input tokens vs entities #66

Open Deeksha-agg04 opened 1 year ago

Deeksha-agg04 commented 1 year ago

Hi @koaning

Can we add a feature where the attention heatmaps can be displayed in the form input tokens on one of the axis and output predicted entities on the other axis of the heatmap?

koaning commented 1 year ago

I'm no longer employed at Rasa and am no longer contributing to this repository.

Deeksha-agg04 commented 1 year ago

@dingusagar @tabergma are you contributing to this repo?