ailia-ai / ailia-ai/ailia-models

ADD comet

オープン
#1,843 コメント 0 件 リアクション 0 件 担当者 0 名 GitHub で見る
主要言語
Python
スター
2.4k
フォーク
365
平均マージ
3日 17時間
マージ済み PR(30日)
13

説明

翻訳評価モデル。
https://github.com/Unbabel/COMET
https://tbest.hatenablog.com/entry/2025/03/13/101735

```
# インストール
# pip install transformers==4.42.4
# pip install unbabel-comet

from comet import download_model, load_from_checkpoint

# モデルのダウンロードと読み込み
model_path = download_model("Unbabel/wmt22-comet-da")
model = load_from_checkpoint(model_path)

# データ: src(原文), mt(機械翻訳), ref(参照訳) の3つ組
data = [
{
"src": "The fire could be controlled.",
"mt": "火災は制御することができた。",
"ref": "火災を抑えることができた。"
},
{
"src": "Schools and kindergartens were opened.",
"mt": "学校や幼稚園が開校された。",
"ref": "学校と幼稚園が開かれた。"
},
]

# 予測(GPUがない場合は gpus=0)
model_output = model.predict(data)#, batch_size=8, gpus=0, num_workers=0)

# 結果
print(model_output.scores) # 各文のスコア(0〜1)
print(model_output.system_score) # システム全体の平均スコア
```

コントリビューションガイド

このリポジトリのコントリビューションガイドは索引されていません

評価

この issue はまだ評価されていません。

新しい issue をメールで受け取る

初心者向けの GitHub issue を短くまとめたダイジェスト。