我已经创建并训练了DNN_Regressor模型。现在,我希望冻结模型并将其转换为可放入Android项目的形式。
现在,我被路障了,因为提供的freeze function要求我输入“输出节点名称”。这是一个问题,因为我使用了DNN_Regressor模板,并且无法控制节点名称,因此我不知道它们。我花了很多时间试图找到一种获取这些名称的方法,到目前为止,我正在使用TensorBoard查看.pb图形文件,但是我仍然无法弄清楚输出节点的名称。
我很确定我只有1个输出节点,因为我有4个数值特征试图预测1个数值输出值。
是否存在我不知道的默认节点输出名称?请帮忙!