def __init__(self, size, Y_vec, n_drop=10, previous_s=None):
    self.previous_s = previous_s
    self.n_drop = n_drop
    super(BALDDropoutStrategy, self).__init__(size, Y_vec)
def __init__(self, size, Y_vec, n_drop=10, previous_s=None):
    self.previous_s = previous_s
    self.n_drop = n_drop
    super(BALDDropoutStrategy, self).__init__(size, Y_vec)