こんちには。 データアナリティクス事業本部 機械学習チームの中村です。 Hugging Faceのライブラリの使い方紹介記事第2弾です。 今回は、学習時にEarly Stoppingを使用する方法をご紹介します。 Early Stoppingとは Deep Learningなどのモデルを学習する際には、通常trainデータとvalidデータ(検定データ)に分割し、 trainデータでモデル自体を更新し、validデータで学習外データに対する汎化性能を確認します。 より具体的にはvalidデータの損失関数などの指標を確認し、最良となるようなepochで学習を停止するなどの操作をします。 これがEarly Stoppingと一般的に呼ばれるものです。 実行環境 今回はGoogle Colaboratory環境で実行しました。 ハードウェアなどの情報は以下の通りです。 GPU: Tesla P1
![Hugging FaceでEarlyStoppingを使用する方法 | DevelopersIO](https://cdn-ak-scissors.b.st-hatena.com/image/square/f699be39a14cf63230a8228dd33c7a271ad097d3/height=288;version=1;width=512/https%3A%2F%2Fdevio2023-media.developers.io%2Fwp-content%2Fuploads%2F2022%2F09%2Fhugging-face-eyecatch.png)