Just a little add-on: if the Huggingface Dataset consists of train and test data, after the set_format you should write for example:
dataset['train'][10]
Otherwise, it returns Invalid key error.
Just a little add-on: if the Huggingface Dataset consists of train and test data, after the set_format you should write for example:
dataset['train'][10]
Otherwise, it returns Invalid key error.