BERTを用いたコロナ関連論文検索システム

kaggleで公開されていたコロナ関連論文のデータセットを用いて、ある論文がコロナに関する内容になっているか分類するモデルを構築した。工夫した点としては、covid-19など、コロナウイルス関連のキーワードを抜いた文章を作成し、それらを正解データとして与えることで、明示的にコロナ関連のワードが入ってない関連論文も見つけられるよう学習した