blather
Train and use generative text models in a few lines of code.
To see blather in action check out the colab notebook!
Installation
Use the package manager pip to install blather.
pip install blather
Usage
from blather import Blather
blather = Blather()
# fine tunes an appropriate model on your dataset
blather.read("example_dataset.txt")
# returns a text sample generated from the model
blather.write('Sample text to complete')
# saves model
blather.save('model.pt')
# load model from previous training
blather.load('model.pt')
Contributing
Pull requests are welcome. For major changes, please open an issue first to discuss what you would like to change.
Please make sure to update tests as appropriate.