/alpaca-lora-create-news-title

Alpaca-LoRAをlivedoorニュースコーパスでFineTuningさせるサンプルコード

Primary LanguagePythonApache License 2.0Apache-2.0

alpaca-lora-create-news-title

Alpaca-LoRAをlivedoorニュースコーパスでFineTuningさせるサンプルコード

環境構築

pip install -r requirements.txt

livedoorニュースコーパスをダウンロードしてdataディレクトリの中に格納してください。

実行手順

学習済みのデータや整形済みのデータも格納しているため、1や2の手順を飛ばして推論だけを行うことも可能です。
学習にはRTX3080で10時間ほどかかりました。

  1. 学習データの作成
python create_alpaca_format.py
  1. 学習
python finetune_livedoor_news.py
  1. 推論
python generate_news_title.py

推論結果

推論結果