使用CNN对CSV文件进行分类

Question

我有一些心电图数据记录。每条记录都是csv格式，包含12行500列。我打算使用2D CNN对这些文件进行分类。我需要将这些csv文件转换为图像格式吗？我想到了使用keras的imagedatagenerator。可以使用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)