Please I am very confused, I need to train the code on my dataset, but I don't have Idea how can modify in the code