Metadata-Version: 2.1
Name: next-word-prediction
Version: 0.1.0
Summary: Experiment on the Generative Pretrained Transformer 2 (GPT-2) for Language Modeling task using the PyTorch-Transformers library.
Home-page: https://github.com/rdgozum/Next-Word-Prediction
Author: Ryan Paul Gozum
Author-email: ryanpaul.gozum@gmail.com
License: MIT
Description: # Next Word Prediction
        Experiment on the Generative Pretrained Transformer 2 (GPT-2) for Language Modeling task using the PyTorch-Transformers library.
        
        ## Installation
        Requires python>=3.5, pytorch>=1.6.0, pytorch-transformers>=1.2.0
        ```bash
        pip install next-word-prediction==0.1.5 pytorch-transformers==1.2.0 torch==1.6.0
        ```
        
        ## How to use
        ```python
        >>> from next_word_prediction import GPT2
        >>> gpt2 = GPT2()
        >>> text = "The course starts next"
        >>> gpt2.predict_next(text)
        The course starts next ['week', 'to', 'month', 'year', 'Monday']
        ```
        
Keywords: language model
Platform: UNKNOWN
Classifier: License :: OSI Approved :: MIT License
Classifier: Programming Language :: Python :: 3.6
Classifier: Programming Language :: Python :: 3.7
Classifier: Programming Language :: Python :: 3.8
Description-Content-Type: text/markdown
