首页 > 解决方案 > 如何在 Keras 中作为数组推断后获取输出

问题描述

我正在尝试将数据值传递给经过训练的模型进行推理。推理后,我想将输出值保存为数组而不是框架。我怎样才能在 Keras 中做到这一点?这是步骤,我现在正在做;

第一步:加载一个需要推断的样本值。

x = data[0]; //Shape of data[0] is (1,480)

第 2 步:加载经过训练的 Keras 模型。

model = keras.models.load_model('model.h5')

第 3 步:从经过训练的模型中推断数据。

result_before_save = model(x)

如果我在 Spyder 的控制台上打印,我可以看到模型的输出,因为print(aa=result_before_save[0][1]) 形状(2,64,64)与模型要输出的形状相同。但是,当我aa在变量资源管理器中看到时,整个框架看起来像这样: 这是一张图片

标签: pythonkerastf.keras

解决方案


推荐阅读