That is a solution for feature-classifier heatmap and perhaps what you read. For sample-classifier heatmap see this solution:
feature-classifier heatmap
sample-classifier heatmap