transformer とrinnaで作るチャットボットのinput_idsについてです

Question

前提・実現したいこと

現在、transformerとrinnaでチャットボットを作っております。これをFlaskとHerokuを利用してLINEで使えるようにしようとしている最中なのですが、ここで疑問が出てきました

発生している問題

input_message = "今日の天気は？"

input_ids = tokenizer.encode(
    "私: " + input_message + "\nAI: ",
    return_tensors="pt"
)

ここの"私:"と"\nAI:"の部分を、例えば自分の名前などに変更したらどういう影響がありますか？
目に見えない部分なのでわかりません・・下記のURLのサイトの手順2の部分に役割は書いてありますが・・これを記述することによって、これは対話文だということをAIが理解し返答が対話文らしくなるということでしょうか？

該当のソースコード

https://axross-recipe.com/recipes/287#chapter-19128
を参考に作っておりまして、今回の疑問はここの手順2の部分にあります。この部分はほぼコピペです

試したこと

実際に名前を入力してみました。ですが実行結果だけではどう変化したのかはわかりませんでした

Answer 1

AIの部分を変えるとそれなりに変わるかもしれないし、そうならないかもしれないです。
世の中に存在する文にいかにもありそうなものを入れて試すといいと思いますよ。
Hugging Faceでちょっと試した結果からそれっぽいもの（だけ）を集めるとこんな感じです。

私: 明日晴れますか機長: はい、機敏に動いてくれます。おかげさまで着陸態勢に入りました。今夜もありがとうございました。 私 「そんなこともわかってますから。

私: 具合はどうですか医者: えっ、それって薬? 何ですか? 私: ええ、でも、それだとお湯がとれないですね・・・。 私: え?

私: 具合はどうですか技師: 私たちの工場、この工場、どうやってあなたたちの仕事を売って売ったのですか?私にはわかりません。 なぜ工場と私を知っているのか。それはあなたがいつも彼ら

Comments

[Soukyuu777]

ご回答ありがとうございます！
私:や\nAIを記述することによって、返答が対話文らしくなるということでしょうか？

[quickquip]

\n がトークンとして処理されているかはちょっとわかりませんが、AIの部分はそういう感じで働く（はず）と思ってます。