Керас: Как проверить, какой тип данных ожидается на входном слое?

Question

Я хочу проверить, какой тип данных по умолчанию ожидается на входном слое и как его изменить?

Answer 1

Если у вас есть модель keras, просто выполните:

print(model.inputs[0].dtype

Вы можете передать желаемый тип данных при создании модели. Например:

import tensorflow as tf 
print(tf.__version__) # 2.0.1

x = tf.keras.layers.Input((2,), dtype=tf.float64)
res = tf.keras.layers.Dense(2, dtype=tf.float64)(x)
model = tf.keras.models.Model(x, res)

print(model.inputs[0].dtype, model.outputs[0].dtype) # <dtype: 'float64'> <dtype: 'float64'>