search

axinc-ai / ailia-models

The collection of pre-trained, state-of-the-art AI models for ailia SDK
2k stars 318 forks source link

Implement t5_base_japanese_ner #1445

Closed kobakos closed 1 month ago

kobakos commented 5 months ago

https://github.com/axinc-ai/ailia-models/issues/1403

kyakuno commented 1 month ago

モデルアップロード済み。 https://storage.googleapis.com/ailia-models/t5_base_japanese_ner/t5_base_japanese_ner_dec.onnx

kyakuno commented 1 month ago

@kobakos こちら、独自に学習されたモデルということで良いでしょうか？

kyakuno commented 1 month ago

Slackを確認させていただくと、独自に学習されたモデルですね。学習、ありがとうございました。

訓練コードを走らせておいたスパン型のモデルがまだ収束していなかったので追加で20エポックほど走らせるとValidation Lossがようやく減らなくなった。生成した出力を評価するもF値は0.35程度と大きく劣る結果になった。書き抜き型モデルのサンプル作成。