Skip to content

Instantly share code, notes, and snippets.

@shangeth

shangeth/dropout.py

Created December 27, 2018 06:00
Download ZIP

Revisions

  1. shangeth created this gist Dec 27, 2018.
    13 changes: 13 additions & 0 deletions dropout.py
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,13 @@
    def dropout(X, drop_probability):
    keep_probability = 1 - drop_probability
    mask = np.random.uniform(0, 1.0, X.shape) < keep_probability
    if keep_probability > 0.0:
    scale = (1/keep_probability)
    else:
    scale = 0.0
    return mask * X * scale



    x = np.random.randn(3,3)
    dropout(x, 0.3)