Skip to content

a toy demonstration of uncertainty quantification in neural networks by dropout method

Notifications You must be signed in to change notification settings

junfeng-chen/dropout_for_uq

Repository files navigation

dropout_for_uq

a toy demonstration of uncertainty quantification in neural networks by dropout method.

The implementation is based on Keras sequential API. To activate dropout in the predictions, modify the source code of Keras as follows:

In class Dropout(in layers/core.py), function def call(self, inputs, training=None), change "return K.in_train_phase(dropped_inputs, inputs, training=training)" to "return K.in_train_phase(dropped_inputs, dropped_inputs, training=training)"

About

a toy demonstration of uncertainty quantification in neural networks by dropout method

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages