Skip to content

Chinese Word Segmentation task based on BERT and implemented in Pytorch

Notifications You must be signed in to change notification settings

melody-rain/chinese-word-segmentation-pytorch

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

68 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Code style: black

Chinese Word Segmentation

Architecture

The baseline architecture is based on the work of Ma et al.. An improved transformer-based architecture is also implemented.

The dataset can be downloaded from here.

Results

Model AS CITYU MSR PKU
This work 96.5 97.5 97.7 96.3
Ma et al. (2018) 96.2 97.2 97.4 96.1

Run

The train script is in cws/train.py. Run this to see all the input parameters

python cws/train.py -h

You can generate segmented sentences by running cws/predictor.py.

python cws/predictor.py -h

The evaluation uses the official scripts, in scripts/score.

About

Chinese Word Segmentation task based on BERT and implemented in Pytorch

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 68.2%
  • Perl 26.3%
  • Shell 5.5%