I want to view the heatmap like in the paper, But I can't find the way. Someone can help me?. Give me some code or document to do it