# what R and Python package can predict a word list with certain probability after one known word?

I have a lot of documents, I need to know after one know word, which words occurs with which probability. For example:

The one known word is IT, the the following words could be:
company, 30%
market, 23%
product, 17%
......


The whole sentence might or might not be involved in, which I am not sure now. So, I do not need some very huge deep learning algorithms by now. I think some simple neural network or Bayesian can do this. But I just do not remember which R and Python packages could do this.

Could any one recommend some packages for me?