Hub documentation
Hugging FaceでのSpanMarkerの使用方法
Hugging FaceでのSpanMarkerの使用方法
SpanMarkerは、BERT、RoBERTa、DeBERTaなどの馴染みのあるエンコーダーを使用して、強力な名前実体認識モデルを訓練するためのフレームワークです。🤗 Transformersライブラリの上に密接に実装されているため、SpanMarkerはその恩恵を十分に受けることができます。その結果、Transformersに慣れている方であれば、SpanMarkerも直感的に使いこなすことができるでしょう。
HubにおけるSpanMarkerの探索
モデルページの左側でフィルタリングすることで、span_markerモデルを見つけることができます。
Hub上のすべてのモデルには、これらの便利な機能が備わっています:
- 簡潔な説明付きの自動生成されたモデルカード。
- ブラウザ上でモデルを直接操作できるインタラクティブなウィジェット。
- 推論リクエストを行うことを可能にするInference Providersウィジェットです。
インストール
始めるには、SpanMarkerのインストールガイドに従うことができます。また、以下の1行のpipコマンドでインストールすることもできます。
pip install -U span_marker既存のモデルを使用する
すべてのspan_markerモデルは、Hubから簡単に読み込むことができます。
from span_marker import SpanMarkerModel
model = SpanMarkerModel.from_pretrained("tomaarsen/span-marker-bert-base-fewnerd-fine-super")読み込みが完了すると、SpanMarkerModel.predict を使用して推論を実行できます。
model.predict("Amelia Earhart flew her single engine Lockheed Vega 5B across the Atlantic to Paris.")[
{"span": "Amelia Earhart", "label": "person-other", "score": 0.7629689574241638, "char_start_index": 0, "char_end_index": 14},
{"span": "Lockheed Vega 5B", "label": "product-airplane", "score": 0.9833564758300781, "char_start_index": 38, "char_end_index": 54},
{"span": "Atlantic", "label": "location-bodiesofwater", "score": 0.7621214389801025, "char_start_index": 66, "char_end_index": 74},
{"span": "Paris", "label": "location-GPE", "score": 0.9807717204093933, "char_start_index": 78, "char_end_index": 83}
]特定のSpanMarkerモデルを読み込みたい場合は、Use in SpanMarkerをクリックすると、動作するスニペットが表示されます。