Skip to content

sohyeon98720/script_generation_kr

This branch is 1 commit behind generative-model/script_generation_kr:main.

Folders and files

NameName
Last commit message
Last commit date

Latest commit

0e91d94 · Jan 26, 2021

History

38 Commits
Dec 30, 2020
Dec 29, 2020
Dec 29, 2020
Dec 29, 2020
Dec 29, 2020
Dec 29, 2020
Dec 29, 2020
Jan 26, 2021
Dec 29, 2020
Dec 29, 2020
Dec 30, 2020
Dec 29, 2020
Dec 30, 2020

Repository files navigation

script_generation_kr

finetuning

python main.py --epoch=200 --data_file_path=./train.csv --save_path=./checkpoint/ --load_path=./checkpoint/KoGPT2_checkpoint_240000.tar --batch_size=1

generation

download trained model here or train your own model through finetuning

python generator.py --temperature=1.0 --text_size=100 --load_path=./checkpoint/KoGPT2_checkpoint_240000.tar --tmp_sent="우리는 지난"

or

generator_ngram_creativity.ipynb(recommended option)


Korean Bert Score

git clone https://github.com/lovit/KoBERTScore
cd ko-BERTScore
python setup.py install

reference

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 78.7%
  • Python 21.3%