Could you elaborate on what you did when you say “corrected the number of label classes”? As in which at step of the code did you have to change it? What was the incorrect number and what was the correct number?