visualization script for Figure 2

[ErikZ719]

Great work! Could you please explain how to generate the NIAH visualization results shown in Figure 2? Could you also share the visualization script? Thank you very much.

[lxr-tech]

Thank you for your recognition of our work.

The visualization script for the NIAH task can be found at https://github.com/OpenMOSS/LongLLaDA/blob/main/needlebench/needlebench_summarizer.py#L256.

For the complete NIAH evaluation, please refer to https://github.com/OpenMOSS/LongLLaDA?tab=readme-ov-file#needle-in-a-haystack-niah-evaluation. After running the evaluation with OpenCompass, the plot is automatically generated.