问答详情
源自:3-7 TensorFlow结合mnist进行卷积模型训练(4)

在预测的时候keep_prob 不应该为0吗

如题,预测的时候不需要进行drop out 的操作了吧

提问者:我来学py 2018-10-10 13:12

个回答

  • qq_深猫_04306026
    2018-11-28 17:14:05

    预测的时候不需要drop out 的操作。

    tf.nn.dropout里面的参数keep_prob的意思是:留下的神经元的概率,如果keep_prob为0的话, 就是让所有的神经元都失活。

    类似的函数是tf.layers.dropout,里面的参数是rate,是失活的概率。

    可以看https://stackoverflow.com/questions/44395547/tensorflow-whats-the-difference-between-tf-nn-dropout-and-tf-layers-dropout。