hxdaze / alpaca-lora-create-news-title

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

推論結果

推論結果

About

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

License:Apache License 2.0


Languages

Language:Python 100.0%