Esempio n. 1
0
 def _init_distribution(conditions):
     alpha = conditions["alpha"]
     return tfd.Zipf(power=alpha)
Esempio n. 2
0
 def _base_dist(self, alpha: TensorLike, *args, **kwargs):
     return tfd.Zipf(power=alpha, *args, **kwargs)
Esempio n. 3
0
 def _init_distribution(conditions):
     power = conditions["power"]
     return tfd.Zipf(power=power)
Esempio n. 4
0
 def _base_dist(self, alpha: TensorLike, *args, **kwargs):
     return tfd.Zipf(power=alpha, dtype=tf.float32, *args, **kwargs)