search

weizhepei / InstructRAG

InstructRAG: Instructing Retrieval-Augmented Generation via Self-Synthesized Rationales
https://weizhepei.com/instruct-rag-page
MIT License
54 stars 4 forks source link

How to visualize Attention Visualization #4

Closed Se-Hun closed 1 month ago

Se-Hun commented 1 month ago

Hello, thank you for sharing your excellent methodology. As I was reviewing your paper, I have a question.

My Question is: How do you implement attention visualization for the retrieved documents as shown below? Could you share the code or specific approach for it?

weizhepei commented 1 month ago

Thanks for your interest in our work! You can refer to this script for attention visualization. Please let me know if you have further questions!

Se-Hun commented 1 month ago

Thanks for your answering !