Getting a KeyError for the first three index values in a custom dataset class in pytorch but working fine after that

Question

I called data from a csv file and split it into training and testing. The data was later transferred to a the custom dataset class and later loaded in a DataLoader. The trainloa