почему keras "flow_from_dataframe" не работает в ядре kaggle? - PullRequest
Новая
       21

почему keras "flow_from_dataframe" не работает в ядре kaggle?

0 голосов
/ 31 марта 2019

Я пытаюсь настроить классификацию изображений в ядре Kaggle.Модель отлично работает, когда работает на моем ПК (Tf-GPU / Keras под Anaconda).Когда я пытаюсь запустить его в ядре Kaggle, я получаю следующую ошибку:

If class_mode="categorical", y_col="has_cactus" column values must be type string, list or tuple.

Это часть кода, к которой относится сообщение об ошибке: 

df_data = pd.read_csv(train_label_path)

train_gen = datagen.flow_from_dataframe(
                            dataframe= df_data,                                
                            directory= train_img_path,
                            x_col='id',
                            y_col='has_cactus',
                            target_size=(IMAGE_SIZE,IMAGE_SIZE),
                            batch_size= BATCH_SIZE,
                            class_mode='categorical',
                            subset='training'
                            )

результат df_data.head() в моей системе или ядре Kaggle одинаков: 

                id                          has_cactus

0  0004be2cfeaba1c0361d39e2b000257b.jpg           1

1  000c8a36845c0208e833c79c1bffedd1.jpg           1

2  000d1e9a533f62e55c289303b072733d.jpg           1

3  0011485b40695e9138e92d0b3fb55128.jpg           1

4  0014d7a11e90b62848904c1418fc8cf2.jpg           1

мой вопрос: почему flow_from_dataframe делает ошибку только в ядре kaggles?кстати, так же, когда я пробую гугл колаб.

...