This is what happens when I try to split your code into two part, one for training and one for testing. can you please help me out as to how it should be done properly?