Is there a way to get probability of next word?

[orenschonlab]

@rdgozum Is there a way to get probability of specidic next word?
For example,

>>> from next_word_prediction import GPT2
>>> gpt2 = GPT2()
>>> text = "The course starts next"
>>> gpt2.get_next_prob(text, 'semester')
0.024

What will be the best way to implement such function?

[sktguha]

If not does it maybe rank by list of most probable first in the result ?

[rdgozum]

@sktguha The function actually returns k words ranked from highest probability.

[sktguha]

ohh that's nice , thanks. actually one issue I find is it doesn't predict word completion
only full next words are returned. any package you might know that does that ??

Thanks in Advance