Я не могу обучить модель Char-RNN в течение нескольких эпох - я получаю AttributeError: у объекта 'PrefetchDataset' нет атрибута 'ndim'

Question

Я пытаюсь предсказать следующий символ на основе предыдущих 100 символов, используя RNN с 2 слоями GRU по 128 единиц каждый и с 20% -ым выпадением как на входах (выпадение), так и на скрытых состояниях (recurrent_dropout). Я ввел код:

model = keras.models.Sequential([
    keras.layers.GRU(128, return_sequences=True, input_shape=[None, max_id],
                     dropout=0.2, recurrent_dropout=0.2),
    keras.layers.GRU(128, return_sequences=True,
                     dropout=0.2, recurrent_dropout=0.2),
    keras.layers.TimeDistributed(keras.layers.Dense(max_id,
                                                    activation="softmax"))
])
model.compile(loss="sparse_categorical_crossentropy", optimizer="adam")
history = model.fit(dataset, epochs=20)

вывод:

AttributeError Traceback (most recent call last)
AttributeError:'PrefetchDataset' object has no attribute 'ndim'

Answer 1

Во время создания набора данных вы пытаетесь получить доступ к атрибуту с именем ndim после предварительной выборки (вызов .prefetch() для вашего набора данных приводит к экземпляру PrefetchDataset). Возможно, это где-то опечатка, но, не видя ваш код, больше ничего не скажешь.