KBNLresearch / ochre

Toolbox for OCR post-correction
Apache License 2.0
122 stars 18 forks source link

Add missing parameters for seq2seq model #8

Open fuzzythecat opened 6 years ago

fuzzythecat commented 6 years ago

initialize_model_seq2seq was missing a positional parameter pred_char, and running lstm_synced.py with seq2seq = True runs into an error.

This PR adds pred_char plus an optional parameter char_embedding_size for enabling the embedding layer in the seq2seq model.