导入错误：无法从“keras.utils”导入名称“序列”

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()

然而，它在执行时显示错误

<块引用>

导入错误：无法从 'keras.utils' 导入名称 'Sequence' (/usr/local/lib/python3.7/dist-packages/keras/utils/init.py)

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

Answer 1

您可以尝试使用 tf.keras 代替 keras

import tensorflow.keras as keras