from datasets import load_dataset
 
# Dataset id from huggingface.co/dataset
dataset_id = "banking77"
 
# Load raw dataset
raw_dataset = load_dataset(dataset_id)
 
print(f"Train dataset size: {len(raw_dataset['train'])}")
print(f"Test dataset size: {len(raw_dataset['test'])}")