/auto-punctuation-pytorch

Primary LanguageJupyter NotebookMIT LicenseMIT

Using Deep Learning for auto punctuation

This work utilizes a bi-directional GRU to retrieve the punctuations from the text.

For details, please refer to the Jupyter Notebook file:

main.ipynb

Implementation

Simply run the following file:

python main.py

If you desire to check ho the data has been processed, please refer to:

processData.py

Requirements

pytorch
pandas
tqdm
numpy
sklearn