keras - ImportError：无法从“keras.utils”导入名称“Sequence”

Question

我正在尝试在 keras 中创建一个自定义数据生成器 -

class DataGenerator_cnn(keras.utils.Sequence):
'Generates data for Keras'
def __init__(self, list_IDs, labels, batch_size=5, dim=(30,128,157),
             n_classes=1, shuffle=True):
    'Initialization'
    self.dim = dim
    self.batch_size = batch_size
    self.labels = labels
    self.list_IDs = list_IDs
    self.n_classes = n_classes
    self.shuffle = shuffle
    self.on_epoch_end()

但是，它在执行时显示错误

ImportError：无法从“keras.utils”（/usr/local/lib/python3.7/dist-packages/keras/utils/ init .py）导入名称“Sequence”

我在 google colab 上使用最新版本的 keras (2.5.0)。请帮忙

标签： kerastensorflow2.0

Answer 1

您可以尝试使用tf.keraskeras

import tensorflow.keras as keras