I have a question, your Figure 4 looks good, how do you do the model visualization to check?Could you show the code about it?