首页 > 解决方案 > 如何在 Huggingfaces 中从 CSV 加载自定义数据集

问题描述

我想使用从 csv 加载自定义数据集huggingfaces-transformers

标签: huggingface-transformershuggingface-datasets

解决方案


来自https://huggingface.co/docs/datasets/loading_datasets.html#loading-from-local-files

dataset = load_dataset('csv', data_files={'train': "train_set.csv",'test': "test_set.csv"})

推荐阅读