Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
def __init__(self, logits):
"""
Probability distributions from Bernoulli input
:param logits: ([float]) the Bernoulli input data
"""
self.logits = logits
self.probabilities = tf.sigmoid(logits)
super(BernoulliProbabilityDistribution, self).__init__()
def probability_distribution_class(self):
return BernoulliProbabilityDistribution