5-4 keras_to_estimator 为什么output 是2列,而test label的形状是1列

来源:5-4 keras_to_estimator

慕姐9068210

2019-12-16

如题,我打印了 test的形状是1列的,模型设置里面output却是2 softmax,我自己在另一个模型里面试验,两列的输出会报错,报错信息是None,2 与 None,1 形状不匹配,我改成output为1 sigmoid就可以了,请问老师的代码为什么可以work?

写回答

1回答

正十七

2019-12-22

同学你好,相应的代码你可以贴一下么?我看了我们的tf_keras_to_estimator的notebook,只有train 和valid数据集,没有test

0
0

Google老师亲授 TensorFlow2.0 入门到进阶

Tensorflow2.0实战—以实战促理论的方式学习深度学习

1849 学习 · 896 问题

查看课程