Safetensors
distilbert
File size: 165 Bytes
f477cda
 
 
 
1
2
3
4
5
from datasets import load_dataset

data = load_dataset("csv", data_files={"train": "train.csv"})
data = data["train"].train_test_split(test_size=0.1)  # 90/10 split