Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.
You can not select more than 25 topics
Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
ZhangbuDong
350b936691
|
1 year ago | |
|---|---|---|
| img | 1 year ago | |
| README.md | 1 year ago | |
| dataset.py | 1 year ago | |
| env.yml | 1 year ago | |
| gen.py | 1 year ago | |
| main.py | 1 year ago | |
| model.py | 1 year ago | |
README.md
Chinese Text Generation Project
Text generation system based on a mixed corpus of 《毛澤東語錄》(Quotations From Chairman Mao Tse-Tung) and《論語》(Confucian Analects).
Based on Transformer. Reference: PyTorch tutorial
Usage
# Create & activate conda environment
$ conda env create -f env.yml
$ conda activate chinese-text-generation
# Train & test with default params
$ python main.py data/corpus.txt output/model.pt
# Train with other settings, see help
$ python main.py -h
# Generate text for human evaluation
$ python gen.py data/corpus.txt output/model.pt
# Generate text with other settings, see help
$ python gen.py -h
