Commit 876524ec authored by lucas_miranda's avatar lucas_miranda
Browse files

Implemented Montecarlo Dropout custom layer

parent f1f98af4
...@@ -60,6 +60,7 @@ class MCDropout(tf.keras.layers.Dropout): ...@@ -60,6 +60,7 @@ class MCDropout(tf.keras.layers.Dropout):
def call(self, inputs, **kwargs): def call(self, inputs, **kwargs):
return super().call(inputs, training=True) return super().call(inputs, training=True)
class DenseTranspose(Layer): class DenseTranspose(Layer):
def __init__(self, dense, output_dim, activation=None, **kwargs): def __init__(self, dense, output_dim, activation=None, **kwargs):
self.dense = dense self.dense = dense
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment