How to draw the Fig.4？

[Luna-LuZi]

Thank you for your sharing, may I ask how the figure 4 in the paper is drawn? Can you provide the code?

[AFeng-x]

hi, you can check the example in (https://github.com/AFeng-x/SMT/tree/main/visualiza/vis.ipynb).