Better handling of integer versus one hot class labels