0 votes
1 view
in Machine Learning by (12.5k points)

I am using dropout in neural network model in keras. Little bit code is like

model.add(Dropout(0.5))

model.add(Dense(classes))

For testing, I am using preds = model_1.predict_proba(image).

But while testing Dropout is also participating to predict the score which should not be happen. I search a lot to disable the dropout but didn't get any hint yet.

Do anyone have solution to disable the Dropout while testing in keras??

1 Answer

0 votes
by (32.8k points)
edited by

Keras does this by default. In Keras dropout is disabled in test mode. You can use the dropped input in training and the actual input while testing.

You have to build your own training function from the layers and specify the training flag to predict with dropout (e.g. it's not possible to specify a training flag for the predict functions). You want to do GANs, which use the intermediate output for training and also train the network as a whole, due to a divergence between generated training images and generated test images.

Hope this answer helps you!

If you want to know more about Machine Learning then watch this video:

...