python - TypeError : If class_mode ="categorical", y_col 列值必须是字符串、列表或元组类型

Question

在 keras documentation ，提到"categorical": 2D numpy array of one-hot encoded labels. Supports multi-label output.但就我而言，它不支持。

train_generator_1 = train_data_gen.flow_from_dataframe(annotation,
                                                      directory="data",
                                                      target_size=(img_shape,img_shape),
                                                      x_col="Left",
                                                      y_col=['N','D','G','C','A','H','M','O'],
                                                      class_mode='categorical',
                                                      shuffle=False,
                                                      batch_size=batch_size,
                                                      seed=7)

附上标签图片。
我得到的错误是 TypeError: If class_mode="categorical", y_col="['N', 'D', 'G', 'C', 'A', 'H', 'M', 'O']" column values must be type string, list or tuple.

Answer 1

如果您有多个输出，则必须选择 class_mode 作为“原始”。
我在这里找到了 Vijayabhaskar J 的一个很棒的教程。 https://medium.com/@vijayabhaskar96/multi-label-image-classification-tutorial-with-keras-imagedatagenerator-cd541f8eaf24