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
00
相似问题