Bart R3f

🚀 BART R3F

このリポジトリでは、[2021 훈민정음 한국어 음성・自然言語人工知能コンテスト] の会話要約部門に参加したアラ꿍달라꿍チームの会話要約学習サンプルモデルを公開しています。

このモデルは、bart-pretrained モデルに 2021-dialogue-summary-competition リポジトリの R3F を適用し、会話要約タスクを学習したものです。

学習に使用されたデータは、AIHub 한국어 대화요약 のデータです。

🚀 クイックスタート

このモデルは、会話要約タスクに特化しており、以下のようなパラメータが設定されています。

また、以下のようなサンプルテキストを使用して推論が可能です。

{
  "language": ["ko"],
  "tags": ["summarization"],
  "widget": {
    "text": "[BOS]밥 ㄱ?[SEP]고고고고 뭐 먹을까?[SEP]어제 김치찌개 먹어서 한식말고 딴 거[SEP]그럼 돈까스 어때?[SEP]오 좋다 1시 학관 앞으로 오셈[SEP]ㅇㅋ[EOS]"
  },
  "inference": {
    "parameters": {
      "max_length": 64,
      "top_k": 5
    }
  }
}