Autosuggest

Simple text field that gives suggestions for continuing a word after a press. Based on the Transformer architecture and developed in PyTorch. It is a curses-based TUI.

Training

To train download one or more sources of text and run

train.py SOURCE [SOURCES..] --save path/to/model.pt.zip

See train.py --help for more options. Edit src/params/hyperparams.py to edit hyperparameters.

Execution

To execute run

autosuggest.py path/to/model.pt.zip

Dependencies

• PyTorch
• npyscreen