使用 flow_from_dataframe y_col 的正确“值”是什么

Question

我正在用熊猫读取 csv 文件，并给出存储在 colname

colnames=['file', 'label'] 
# Read data from file 
data = pd.read_csv('./Hand_Annotations_2.csv',names=colnames, header=None)
# Preview the first 5 lines of the loaded data 
data.head()

然后，我使用ImageDataGenerator()和flow_fromdataframe()来获取批量数据

train_generator=datagen.flow_from_dataframe(dataframe=data,
                                            directory=None, 
                                            x_col=colnames[0], 
                                            y_col=colnames[1],
                                            class_indices=IDmap,
                                            class_mode="categorical", target_size=(224,224), batch_size=32)

但我收到一个错误，如下所示：

类型错误：如果 class_mode="categorical"，y_col="label" 列值必须是字符串、列表或元组类型。

但我的y_col是一个字符串。如果我只是输入，我会得到同样的错误"label"。它似乎也适用于x_col.

有人可以指出我的错误吗？

谢谢

---

解决方案

使用 dtype 例如 str 读取 csv：

data = pd.read_csv('./Hand_Annotations_2.csv',dtype=str,names=colnames, header=None)

Answer 1

您可以检查 github 中的代码以了解实际检查的内容，特别是这里。Keras 预处理正在检查所有值是否具有允许的类型（str、list 或 tuple），因此检查失败，您必须检查y_col列的所有值是否具有正确的类型。