from datasets import load_dataset
 
# Load dataset from the hub
dataset = load_dataset("samsum")
 
print(f"Train dataset size: {len(dataset['train'])}")
print(f"Test dataset size: {len(dataset['test'])}")
 
# Train dataset size: 14732
# Test dataset size: 819