the dataset split into validation and training set.i use a mobilenet model. i get an accuracy of 97%. but the confusion matrix is only showing 16. i dont know how to code for co