I think you’re using class_mode=”categorical” while you should be using class_mode=”other”