サブロウ丸

Sabrou-mal サブロウ丸

主にプログラミングと数学

Troubleshooting ImportError: cannot import name 'load_metric' from 'datasets'

from datasets import load_dataset, load_metric
>>> ImportError: cannot import name 'load_metric' from 'datasets'

について、datasetsのバージョン3.0.0のリリースノートにあるようにこのバージョンからload_metric がdatasets から取り除かれました、その代わり evaluate パッケージを使ってね、ということです。

Release 3.0.0 · huggingface/datasets · GitHub

  • Remove metrics by @albertvillanova in #6983
    • remove deprecated load_metric, please use the evaluate library instead

解決策1 evaluateパッケージに入れ替え

# datasets.load_metrics('rouge') 
import evaluate
evaluate.load('rouge')

解決策2 datasets パッケージのダウングレード

datasetsのバージョン2.11.0をインストール。

pip install datasets==v2.11.0

datasets · PyPI

バージョン履歴はPyPI で調べることができます。