Updated code to properly print confusion matrix. And noted where to change numbers for different numbers to train on.
Ran another version to correct how confusion matrix is presented. Accuracy is still bad. Latest run, 7 is predicted better than 5.