I am using logistic regression to classify images with 3 classes and there may be more. My label column is train_data["label"] it consists of strings
train_data["label"]