Add visualization method for language models

[adrhill]

A visualisation method such as plot_text_heatmap from this iNNvestigate notebook would be useful for language tasks such as sentiment analysis.