Uie Base

🚀 UIE情報抽出モデル（Pytorch）

このモデルは、様々な情報抽出タスクを統一的に扱うことができ、異なるタスク間での移行と汎化能力に優れています。また、零サンプルでの迅速な冷スタートや小サンプルでの微調整が可能で、特定の抽出目標にも素早く適応できます。

🚀 クイックスタート

モデルの紹介

• UIE(Universal Information Extraction)：Yaojie LuらはACL - 2022で、汎用情報抽出の統一フレームワーク UIE を提案しました。
• このフレームワークは、エンティティ抽出、関係抽出、イベント抽出、感情分析などのタスクを統一的にモデリングし、異なるタスク間で良好な移行と汎化能力を備えています。
• 皆さんが UIE の強力な機能を簡単に利用できるように、PaddleNLP はこの論文の方法を参考に、ERNIE 3.0 知識強化事前学習モデルに基づいて、最初の中国語汎用情報抽出モデル UIE を訓練してオープンソース化しました。
• このモデルは、業界や抽出対象を限定せずに重要情報を抽出でき、零サンプルでの迅速な冷スタートを実現し、優れた小サンプル微調整能力を備えており、特定の抽出対象にすばやく適応できます。

💻 使用例

基本的な使用法

from transformers import AutoModel, AutoTokenizer

tokenizer = AutoTokenizer.from_pretrained("xusenlin/uie-base", trust_remote_code=True)
model = AutoModel.from_pretrained("xusenlin/uie-base", trust_remote_code=True)

schema = ["時間", "選手", "試合名"]  # Define the schema for entity extraction
print(model.predict(tokenizer, "2月8日上午北京オリンピック冬季大会フリースタイルスキー女子ビッグアイアン決勝で中国の選手谷愛凌が188.25点で金メダルを獲得！", schema=schema))

schema = {'競技名': ['主催者', '実施者', '開催回数']}  # Define the schema for relation extraction
model.set_schema(schema)
print(model.predict(tokenizer, "2022言語と知能技術コンテストは中国中文情報学会と中国計算機学会が共同主催し、百度会社、中国中文情報学会評価委員会と中国計算機学会自然言語処理専門委員会が実施し、4回連続で開催され、世界で最も人気のある中国語NLPコンテストの一つになりました。"))

schema = {'地震トリガー語': ['地震強度', '時間', '震央位置', '震源深度']}  # Define the schema for event extraction
model.set_schema(schema)
print(model.predict(tokenizer, "中国地震台網が正式に発表：5月16日06時08分に雲南省臨滄市鳳慶県(北緯24.34度、東経99.98度)で3.5マグニチュードの地震が発生し、震源深度は10キロメートルです。"))

schema = {'評価次元': ['意見語', '感情傾向[ポジティブ、ネガティブ]']}  # Define the schema for opinion extraction
model.set_schema(schema)
print(model.predict(tokenizer, "店は清潔で静かで、スタッフのサービスも親切で、コスパが非常に高いです。レジで並んでいるのが見られました。"))

schema = "感情傾向[ポジティブ、ネガティブ]"  # Define the schema for opinion extraction
model.set_schema(schema)
print(model.predict(tokenizer, "この製品は本当に使いやすく、私はとても気に入っています。"))

schema = ['裁判所', {'原告': '委任代理人'}, {'被告': '委任代理人'}]  # Define the schema for opinion extraction
model.set_schema(schema)
print(model.predict(tokenizer, "北京市海淀区人民裁判所\n民事判決書\n(199x)建初字第xxx号\n原告：張三。\n委任代理人李四、北京市 A弁護士事務所弁護士。\n被告：B会社、法定代表人王 五、開発会社総経理。\n委任代理人趙六、北京市 C弁護士事務所弁護士。"))

📄 ライセンス

このモデルは apache - 2.0 ライセンスの下で提供されています。

📚 ドキュメント

PaddleNLP