I am trying to run this code in my VS code
from datasets import load_dataset
imdb = load_dataset("imdb")
split = imdb["train"].train_test_split(.2, seed=3463)
imdb["train"] = split["train"]
imdb["val"] = split["test"]
del imdb["unsupervised"]
But I am getting the below error
Python version 3.11.0, datasets version 2.17.1
Can anyone help with this?