- Notifications
You must be signed in to change notification settings - Fork0
Imageboard bot with recurrent neural network (RNN, GRU)
License
NotificationsYou must be signed in to change notification settings
oupirum/sp2cp
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
Бот с нейросетью, обучающейся генерировать комментарии на датасете из тредов с 2ch.hk
https://drive.google.com/drive/folders/1Xmm5DTxQ8sJwL2ygefWTPFWsTSviMPDP
python3 ./grub_threads.py --board=bpython3 ./parse_dataset.py --lexicon_limit=100000 --sequence_max_len=50It will save tokenized dataset and id2token map in./dataset/parsed/ directory.
python3 ./train.py --epochs=20 --batch_size=100After training you can find trained models and id2token map in./models/ directory.
python3 ./autoposter.py --weights_file=./models/weights.h5 --id2token_file=./models/id2token.json \ --board=b --passcode=passcode --max_threads=30 \ --max_res_len=30