我使用数据帧中的流来处理具有 14 个可能标签的多标签分类问题,所有列名称都以字符串格式放置在列表中,例如:
columns = ["No Finding", "Enlarged Cardiomediastinum", "Cardiomegaly", "Lung Opacity", "Lung Lesion","Edema", "Consolidation", "Pneumonia", "Atelectasis", "Pneumothorax", "Pleural Effusion", "Pleural Other", "Fracture", "Support Devices"]
Run Code Online (Sandbox Code Playgroud)
然后将列表名称(列)输入 y_col 例如:
train_generator=datagen.flow_from_dataframe(
dataframe=df[:178731],
directory='/home/admin1/Downloads/',
x_col='Path',
y_col=columns,
batch_size=batch_size,
seed=42,
shuffle=True,
target_size=(224, 224))
Run Code Online (Sandbox Code Playgroud)
我收到此错误:
TypeError: If class_mode="categorical", y_col="['No Finding', 'Enlarged Cardiomediastinum', 'Cardiomegaly', 'Lung Opacity', 'Lung Lesion', 'Edema', 'Consolidation', 'Pneumonia', 'Atelectasis', 'Pneumothorax', 'Pleural Effusion', 'Pleural Other', 'Fracture', 'Support Devices']" column values must be type string, list or tuple.
Run Code Online (Sandbox Code Playgroud)
我已经尝试过之前提出的解决方案:
df['No Finding'] = df['No Finding'].astype(str)
df['Enlarged Cardiomediastinum'] …Run Code Online (Sandbox Code Playgroud) keras ×1