Skip to content

sertiscorp/thai-word-segmentation

Repository files navigation

Thai word segmentation with bi-directional RNN

This is code for preprocessing data, training model and inferring word segment boundaries of Thai text with bi-directional recurrent neural network. The model provides precision of 98.94%, recall of 99.28% and F1 score of 99.11%. Please see the blog post for the detailed description of the model.

Requirements

  • Python 3.4
  • TensorFlow 1.4
  • NumPy 1.13
  • scikit-learn 0.18

Files

  • preprocess.py: Preprocess corpus for model training
  • train.py: Train the Thai word segmentation model
  • predict_example.py: Example usage of the model to segment Thai words
  • saved_model: Pretrained model weights
  • thainlplib/labeller.py: Methods for preprocessing the corpus
  • thainlplib/model.py: Methods for training the model

Note that the InterBEST 2009 corpus is not included, but can be downloaded from the NECTEC website.

Usage

To try the prediction demo, run python3 predict_example.py. To preprocess the data, train the model and save the model, put the data files under data directory and then run python3 preprocess.py and python3 train.py.

Bug fixes and updates

  • 3/10/2019: Switched license to MIT
  • 1/6/2018: Fixed bug in splitting data incorrectly in preprocess.py. The model was retrained achieving precision 98.94, recall 99.28 and F1 score 99.11. Thank you Ekkalak Thongthanomkul for the bug report.
  • 1/6/2018: Load the model variables with signature names in predict_example.py.

Contributors

  • Jussi Jousimo
  • Natsuda Laokulrat
  • Ben Carr
  • Ekkalak Thongthanomkul
  • Vee Satayamas

License

MIT

Copyright (c) Sertis Co., Ltd., 2019

About

Thai word segmentation with bi-directional RNN

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages