Керас: Неспособность указать аргумент шагов

У меня есть модель keras, обученная на tf.data.Dataset, например:

dataset = dataset.map(preprocess)
sess = tf.Session()
sess.run(tf.tables_initializer())
tf.keras.backend.set_session(sess)
dataset = dataset.batch(50).repeat() 
model.fit(dataset, steps_per_epoch=100, epochs=20)

Итак, я обрабатываю таким же образом:

x = tf.data.Dataset.from_tensor_slices(data_test)
x = x.map(lambda x: ({'reviews': x}))
x = x.map(preprocess)

а потом делаю:

preds = self.model.predict(x)

но, к сожалению, вернись:

ValueError: When using iterators as input to a model, you should specify the `steps` argument.

Добавление x = x.batch(50).repeat() перед предсказанием приводит к той же ошибке.

Есть мысли о том, что я делаю неправильно?

Спасибо!

Похоже, что в вашем методе подгонки step_per_epoch дает количество пакетов, составляющих эпоху. Точно так же вам нужно будет указать количество пакетов в тестовом наборе с шагами. Так что, возможно, вы можете объединить тестовый набор данных с размером пакета 1, если вы хотите прогноз для каждой точки данных, а затем выполните что-то вроде model.predict (x .make_one_shot_iterator (), steps = no_of_data_points)?

kvish 31.10.2018 03:36
Почему в Python есть оператор "pass"?
Почему в Python есть оператор "pass"?
Оператор pass в Python - это простая концепция, которую могут быстро освоить даже новички без опыта программирования.
Некоторые методы, о которых вы не знали, что они существуют в Python
Некоторые методы, о которых вы не знали, что они существуют в Python
Python - самый известный и самый простой в изучении язык в наши дни. Имея широкий спектр применения в области машинного обучения, Data Science,...
Основы Python Часть I
Основы Python Часть I
Вы когда-нибудь задумывались, почему в программах на Python вы видите приведенный ниже код?
LeetCode - 1579. Удаление максимального числа ребер для сохранения полной проходимости графа
LeetCode - 1579. Удаление максимального числа ребер для сохранения полной проходимости графа
Алиса и Боб имеют неориентированный граф из n узлов и трех типов ребер:
Оптимизация кода с помощью тернарного оператора Python
Оптимизация кода с помощью тернарного оператора Python
И последнее, что мы хотели бы показать вам, прежде чем двигаться дальше, это
Советы по эффективной веб-разработке с помощью Python
Советы по эффективной веб-разработке с помощью Python
Как веб-разработчик, Python может стать мощным инструментом для создания эффективных и масштабируемых веб-приложений.
2
1
2 393
0

Другие вопросы по теме