katsura-jp / generate-syosetu-title

なろう小説風のタイトルをTransformerによって生成

なろう風タイトルジェネレータ

環境構築

git clone https://github.com/katsura-jp/generate-syosetu-title.git
cd generate-syosetu-title
pip install -r requirements.txt
mkdir model

学習済みモデルはGoogle Driveにアップロードしているのでダウンロードして使ってください（バージョンは1つでも全部でも）. ダウンロードしたzipファイルを作ったmodelディレクトリに移動させて解凍してください．

modelディレクトリは任意の名前で問題ないので，嫌な人は別名でどうぞ．さらに言えば，学習済みモデルへのパスは任意に指定できるのでダウンロード先で解凍してそこを指定してもOKです．

使い方

（例）modelにversion_1を入れ，input.txtにあらすじを記述した場合．また，ビームサーチのサイズに5を指定．

python generate.py --ckpt ./model/version_1/ -i ./input.txt -b 5