Request for visualization code to plot learned super tokens (Figure 4)

[codyshen0000]

Hello! I was going through your paper and I'm interested in reproducing the visualization of the learned super tokens as shown in Figure 4. Could you please provide some guidance or share the code you used to generate that figure? I look forward to your response. Thank you!

[hhb072]

Please refer to the visualization folder.