smash.factory.Net.add_dropout

Net.add_dropout(drop_rate)

Add a dropout layer that randomly sets the output of the previous layer to zero with a specified probability.

Parameters:

drop_ratefloat

The probability of setting a given output value to zero.

Examples

>>> from smash.factory import Net
>>> net = Net()
>>> net.add_dense(1024, input_shape=56)
>>> net.add_dropout(0.25)
>>> net
+------------------------------------------------+
| Layer Type  Input/Output Shape  Num Parameters |
+------------------------------------------------+
| Dense       (56,)/(1024,)       58368          |
| Dropout     (1024,)/(1024,)     0              |
+------------------------------------------------+
Total parameters: 58368
Trainable parameters: 58368