Skip to content

sohyeon98720/script_generation_kr

 
 

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%