next word prediction

次の単語予測

Pytorch-Transformers Libraryを使用した言語モデリング用の生成前処理トランス2（GPT-2）。

インストール

python> = 3.5、pytorch> = 1.6.0、pytorch-transformers> = 1.2.0が必要です

pip install next-word-prediction

使い方

 > >> from next_word_prediction import GPT2
> >> gpt2 = GPT2 ()
> >> text = "The course starts next"
> >> gpt2 . predict_next ( text , 5 )
The course starts next [ 'week' , 'to' , 'month' , 'year' , 'Monday' ]

Streamlitを介したデモ

streamlit run app/run.py