from random import randrange random_id = randrange(len(raw_dataset['train'])) raw_dataset['train'][random_id] # {'text': "I can't get google pay to work right.", 'label': 2}