python - 无法从 HuggingFace 包加载数据集

Question

我正在尝试AMI从HuggingFace 数据集包中加载数据集

from datasets import load_dataset, load_metric
dataset = load_dataset('ami','headset-single')

但是，我收到以下错误。我究竟做错了什么？

Couldn't find file locally at ami/ami.py, or remotely at https://raw.githubusercontent.com/huggingface/datasets/1.8.0/datasets/ami/ami.py.
The file was picked from the master branch on github instead at https://raw.githubusercontent.com/huggingface/datasets/master/datasets/ami/ami.py.
---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
<ipython-input-21-119884e81101> in <module>()
      1 from datasets import load_dataset, load_metric
----> 2 dataset = load_dataset('ami','headset-single')

11 frames
/usr/local/lib/python3.7/dist-packages/datasets/tasks/__init__.py in task_template_from_dict(task_template_dict)
     25         return None
     26     template = NAME2TEMPLATE.get(task_name)
---> 27     return template.from_dict(task_template_dict)

AttributeError: 'NoneType' object has no attribute 'from_dict'

标签： pythondatasethuggingface-transformershuggingface-datasets