Metadata-Version: 2.1
Name: keras-gpt-2
Version: 0.17.0
Summary: GPT-2
Home-page: https://github.com/CyberZHG/keras-gpt-2
Author: CyberZHG
Author-email: CyberZHG@users.noreply.github.com
License: MIT
Description: # Keras GPT-2
        
        [![Version](https://img.shields.io/pypi/v/keras-gpt-2.svg)](https://pypi.org/project/keras-gpt-2/)
        ![License](https://img.shields.io/pypi/l/keras-gpt-2.svg)
        
        \[[中文](https://github.com/CyberZHG/keras-gpt-2/blob/master/README.zh-CN.md)|[English](https://github.com/CyberZHG/keras-gpt-2/blob/master/README.md)\]
        
        Load pretrained weights and predict with [GPT-2](https://d4mucfpksywv.cloudfront.net/better-language-models/language_models_are_unsupervised_multitask_learners.pdf).
        
        ## Install
        
        ```bash
        pip install keras-gpt-2
        ```
        
        ## Demo
        
        ```python
        import os
        from keras_gpt_2 import load_trained_model_from_checkpoint, get_bpe_from_files, generate
        
        
        model_folder = 'xxx/yyy/117M'
        config_path = os.path.join(model_folder, 'hparams.json')
        checkpoint_path = os.path.join(model_folder, 'model.ckpt')
        encoder_path = os.path.join(model_folder, 'encoder.json')
        vocab_path = os.path.join(model_folder, 'vocab.bpe')
        
        
        print('Load model from checkpoint...')
        model = load_trained_model_from_checkpoint(config_path, checkpoint_path)
        print('Load BPE from files...')
        bpe = get_bpe_from_files(encoder_path, vocab_path)
        print('Generate text...')
        output = generate(model, bpe, ['From the day forth, my arm'], length=20, top_k=1)
        
        # If you are using the 117M model and top_k equals to 1, then the result will be:
        # "From the day forth, my arm was broken, and I was in a state of pain. I was in a state of pain,"
        print(output[0])
        ```
        
Platform: UNKNOWN
Classifier: Programming Language :: Python :: 3
Classifier: License :: OSI Approved :: MIT License
Classifier: Operating System :: OS Independent
Description-Content-Type: text/markdown
