Skip to content

Instantly share code, notes, and snippets.

View toppylawz's full-sized avatar

Adeolu Olofintuyi toppylawz

  • TechWagon
  • Nigeria
View GitHub Profile
# ImageDataGenerator flow_from_dataframe
df_train = pd.read_csv(home_path + r'/emergency_train.csv')
df_train['emergency_or_not'] = df_train['emergency_or_not'].astype('str') # requires target in string format
train_generator_df = datagen.flow_from_dataframe(dataframe=df_train,
directory=home_path+'/images/',
x_col="image_names",
y_col="emergency_or_not",
class_mode="binary",