只加载huggingface datasets数据集的前10条并打印

hehui0921

已于 2024-02-27 10:39:00 修改

阅读量167

点赞数 1

分类专栏： huggingface 文章标签： java 前端服务器

于 2024-02-27 09:12:42 首次发布

本文链接：https://blog.csdn.net/oHeHui1/article/details/136313500

版权

huggingface 专栏收录该内容

33 篇文章 0 订阅

订阅专栏

以下可以正常运行

from datasets import load_dataset
data = load_dataset("rotten_tomatoes", split="train[:10]")
print(data)
for item in data:
    print(item["text"]+" : "+ str(item["label"]))

以下可以正常运行

from datasets import load_dataset
data = load_dataset("fka/awesome-chatgpt-prompts", split="train[:5]")
print(data)
for item in data:
    print(item["prompt"])

from datasets import load_dataset
# data = load_dataset("fka/awesome-chatgpt-prompts", split="train[:5]")
data = load_dataset("squad", split="train[:1]")

print(data)
for item in data:
    print(item["context"])