你的代码应该输入参数logits=ZL(最后一层的输出,不需要激活参数AL,自动计算softmax),以及参数labels=Y,然后自动计算出cost。
注意:
输入的ZL和Y参数应该保持shape类型为(样本数,分类数),若设置为(分类数,样本数)否则代价cost会成本增大:参考官方说明:https://tensorflow.google.cn/api_docs/python/tf/nn/softmax_cross_entropy_with_logits。
你的代码应该输入参数logits=ZL(最后一层的输出,不需要激活参数AL,自动计算softmax),以及参数labels=Y,然后自动计算出cost。
注意:
输入的ZL和Y参数应该保持shape类型为(样本数,分类数),若设置为(分类数,样本数)否则代价cost会成本增大:参考官方说明:https://tensorflow.google.cn/api_docs/python/tf/nn/softmax_cross_entropy_with_logits。