从csv准备数据以在图像分类中进行多类和多标签分类

Question

我具有以下csv文件的结构...

image_id, class, sub_class, label1, label2, label3, color   
34453572, big, rectangle, 1, 0, 1, red    
42524543, small, circle, 1, 1, 0, black

我有以下python代码：

def read_training_data():
    df = pd.read_csv('train.csv', delimiter=',')
    labels = ['class', 'sub_class', 'label1', 'label2', 'label3', 'color']
    datagen = ImageDataGenerator(rotation_range=30, horizontal_flip=True)
    train_generator = datagen.flow_from_dataframe(dataframe=df, directory="train_set", x_col="tag_id", y_col=labels, has_ext=False, class_mode="categorical", target_size=(64, 96), batch_size=32)
    return train_generator

当参数y_col ='something'（单个类或标签分类）时，它起作用
但是当y_col =标签时，会出现以下错误error
有没有什么办法解决这一问题？还是还有其他准备数据的方法...