классификация файлов csv с использованием CNN

Question

У меня есть записи данных ЭКГ. каждая запись находится в формате CSV с 12 строками и 500 столбцами. Я планирую классифицировать эти файлы с помощью 2D CNN. Нужно ли мне конвертировать эти файлы csv в формат изображения? Подумал об использовании генератора изображений кераса. будет ли работать с CSV-файлами?

Answer 1

да, работает

datagen = ImageDataGenerator(    

featurewise_center=True,

featurewise_std_normalization=True,

rotation_range=20,

width_shift_range=0.2,

height_shift_range=0.2,

horizontal_flip=True)

datagen.fit(x_train)

соответствует модели на пакетах с увеличением данных в реальном времени:

model.fit(datagen.flow(x_train, y_train, batch_size=32),

      steps_per_epoch=len(x_train) / 32, epochs=epochs)