Для обучения модели нужно вызвать команду:
python3 train.py --input-dir ./data/anekdoty.txt --model model.pt
Для генерации текста нужно вызвать:
python3 generate.py --model model.pt --input-dir ./data/anekdoty.txt --prefix "приходит вовочка в" --length 50
Пример вывода:
приходит вовочка в ним поймал говорит скоро кто нужно часы продавец да он не знаю а кто у меня уже дурак кончил в уме революционера компьютер кто мужик и спрашивает помню надувную и не сделал кофе просто вчера царь девушек коробка раз тот себе родила тела с руках говорит несмотря в машине с