Modifying code for multi class

[vineeth2309]

Would changing the number of classes in the model compile from 1 automatically retrieve the labels for the other classes, or the code for label preparation would also need changing?

[IAMShashankk]

@tom-roddick @vineeth2309 Do you have any feedback on this?