ortografix 0.2.0

Seq2seq model with attention for automatic orthographic simplification

ortografix

Welcome to ortografix, a seq2seq model for automatic ortografic simplification, coded with pytorch 1.4.

Install

via pip:

pip3 install ortografix

or, after a git clone:

python3 setup.py install

Train

To train a model, run:

ortografix train \
--data /abs/path/to/training/data \
--model-type gru \
--shuffle \
--hidden-size 256 \
--num-layers 1 \
--bias \
--dropout 0 \
--learning-rate 0.01 \
--epochs 10 \
--print-every 100 \
--use-teacher-forcing \
--teacher-forcing-ratio 0.5 \
--output-dirpath /abs/path/to/output/directory/whereto/save/model \
--with-attention \
--character-based

Test

To (qualitatively) evaluate the output of training on a set of 10 randomly selected sentences, run:

ortografix evaluate \
--data /abs/path/to/test/data \
--model /abs/path/to/model/directory/ \
--random 10