I have been working on a question answering model, where I receive answers on my questions by my word embedding model BERT. But I really want to plot something like this: