辍学版本Google Colab出现问题

时间:2019-03-04 15:38:21

标签: tensorflow keras google-colaboratory dropout

我正在尝试添加一个辍学图层,但收到此警告

WARNING:tensorflow:From /usr/local/lib/python3.6/dist-packages/keras/backend/tensorflow_backend.py:3445: calling dropout (from tensorflow.python.ops.nn_ops) with keep_prob is deprecated and will be removed in a future version.
Instructions for updating:
Please use `rate` instead of `keep_prob`. Rate should be set to `rate = 1 - keep_prob`.

我不知道我使用的是rate还是keep_prob。 我在辍学层中的实现如下:

x = Dropout(rate=0.20)(x)

我使用tensorflow后端导入keras。 版本是:

Keras 2.2.4

tensorflow 1.13.1

如何检查是否将0.20用作rate或keep_prob?

预先感谢

2 个答案:

答案 0 :(得分:1)

根据keras文档

keras.layers.Dropout(rate, noise_shape=None, seed=None)

因此,使用

x = Dropout(0.20)(x)

这里是比率0.20。

答案 1 :(得分:-1)

它警告您不要使用keep_prob。但是由于您使用的是速率,所以将是速率,而不是keep_prob。这是一个令人困惑的警告。要检查它是否设置为1,如果知道,则为keep_prob,如果不是,则为辍学率!