"""Loading the dataset and printing out an example """ fine_tune_ds = load_dataset("fancyzhx/amazon_polarity") for elem in fine_tune_ds['train']: print(elem) break