文本分类如何可视化句子各单词对于最终结果的影响程度？

目前好多论文都会可视化注意力。文本分类可视化分析时经常把与特定标签相关的单词标上颜色，这些是如何实现的？调研过BERTVIZ里面无法实现。实现结果类似下面这种