a/ebooks
1
1
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Add some transformers documentation to README

Browse source
This commit is contained in:
Anthony Wang 2022-02-22 16:17:38 -06:00
parent d191b6204f
commit a4dc9c238b
Signed by: a
GPG key ID: BC96B00AEC5F2D76
1 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
README.md
View file

@ -1,10 +1,10 @@
# ebooks
Fediverse ebooks bot using LSTM neural networks in PyTorch
Fediverse ebooks bot using neural networks
## Usage
First, make sure you have `psycopg2`, `Mastodon.py`, and PyTorch installed. Currently only Mastodon with PostgreSQL is supported, but it should be easy to add support for other platforms and databases.
First, install Python dependencies with `pip install psycopg2 torch transformers datasets Mastodon.py`. Currently only Mastodon with PostgreSQL is supported, but it should be easy to add support for other platforms and databases.
Generate the training data from the Mastodon database using `python data.py -d 'dbname=test user=postgres password=secret'`.
@ -17,3 +17,7 @@ Finally, create a Mastodon application for your bot account and run the bot with
- https://closeheat.com/blog/pytorch-lstm-text-generation-tutorial
- https://trungtran.io/2019/02/08/text-generation-with-pytorch/
- https://huggingface.co/docs/transformers/training
- https://huggingface.co/blog/how-to-generate