Keras 的 model.predict 可以返回字典嗎？

Question

文檔https://keras.io/models/model/#predict說model.predict返回 Numpy 預測數組。 在 Keras API 中，有沒有辦法區分這些數組中的哪些是哪些？ 在TF實現中怎么樣？

在同一頁文檔的頂部，他們說“模型可以使用列表指定多個輸入和輸出”。 如果相反，通過字典，似乎沒有任何問題：

my_model = tf.keras.models.Model(inputs=my_inputs_dict, outputs=my_outputs_dict)

調用model.fit ，相同的文檔說“如果模型中的輸入層被命名，您還可以將字典映射輸入名稱傳遞給 Numpy 數組。”

如果my_output_dict的鍵或my_output_dict中的字典值（層）的my_output_dict附加到my_model.predict(...)的輸出， my_model.predict(...)

如果我使用tf.keras.model.save將模型保存到 TensorFlow 的 saved_model 格式 protobuf tf.serving API 以這種方式工作 - 具有命名的輸入和輸出......

Answer 1

使用my_model.output_names

給定的

my_model = tf.keras.models.Model(inputs=my_inputs_dict, outputs=my_outputs_dict)

從my_model.output_names自己創建dict ，它是按預測順序排列的輸出層的name屬性列表

prediction_list = my_model.predict(my_test_input_dict)
prediction_dict = {name: pred for name, pred in zip(my_model.output_names, prediction_list)}