- model.py - основной скрипт модели. В нём описан класс TextGen, который отвечает за создание триграммной модели и генерации фраз
- learning.py - скрипт, предназначенный для обучения.
- generation.py - скрипт, предназначенный для генерации
- train.txt - "Война и мир", на нём тестировалась модель
Файл learning.py имеет два аргумента
- input_text - обязательный, он должен содержать текст, на котором будет обучаться модель
- --output - можно указать имя файла, в который сохранится модель
Параметры
- file - название файла, из которого нужно загрузить модель
- --length - длина генерируемой строки
- --is_rand - способ генерации
- --seed - можно указать seed для генерации вручную