Enriching language models with graph-based context information to better understand textual data