keras_cat = to_categorical(data['G3'], dtype='uint8') keras_cat