how can i run the main code with my own dataset?

[refuserMa]

when I run the main code with my dataset, it comes to this error:

line 116, in train
targets_fast[idx] = targets[neighbor[random.randint(1,num_neighbor)]]
IndexError: index 5 is out of bounds for dimension 0 with size 5

how to select num_neighbor?is that based on batch_size ,or my dataset's label?