ebooks

Fediverse ebooks bot using neural networks

  1. 1
  2. 2
  3. 3
  4. 4
  5. 5
  6. 6
  7. 7
  8. 8
  9. 9
  10. 10
  11. 11
  12. 12
  13. 13
  14. 14
  15. 15
  16. 16
  17. 17
  18. 18
  19. 19
  20. 20
  21. 21
  22. 22
  23. 23
# ebooks

Fediverse ebooks bot using neural networks

## Usage

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'`.

Next, train the network with `python train.py`, which may take several hours.

Finally, create a Mastodon application for your bot account and run the bot with `python bot.py -i mastodon.instance -t access_token`. Enjoy!

## Resources

- 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