tf.keras.Model.predict и call возвращают другой результат - PullRequest
2 голосов
/ 15 апреля 2019

tf.keras.Model.predict и вызов возврата другого результата

import tensorflow as tf
import numpy as np
tf.set
ipts = tf.keras.Input([2])
x = tf.keras.layers.Dense(10)(ipts)
x = tf.keras.layers.Dropout(0.5)(x)
x = tf.keras.layers.Dense(3)(x)
model = tf.keras.Model(ipts, x)
model.summary()
sess = tf.Session()
sess.run(tf.global_variables_initializer())
y_train = model(tf.ones((2,2)),training=True)
y_test = model(tf.ones((2,2)),training=False)
sess.run(y_train)
sess.run(y_test)
model.predict(np.array([[1.,1.],[1.,1.]]))

sess.run(y_test) должен совпадать с model.predict(np.array([[1.,1.],[1.,1.]])), но факт в том, что они разные.Почему?

1 Ответ

3 голосов
/ 15 апреля 2019

Вам необходимо зарегистрировать сеанс как сеанс Keras с K.set_session(sess). Тогда sess.run(y_test) дает тот же результат, что и model.predict(np.array([[1.,1.],[1.,1.]])).

...