def _init_distribution(conditions):
alpha = conditions["alpha"]
return tfd.Zipf(power=alpha)
def _base_dist(self, alpha: TensorLike, *args, **kwargs):
return tfd.Zipf(power=alpha, *args, **kwargs)
def _init_distribution(conditions):
power = conditions["power"]
return tfd.Zipf(power=power)
def _base_dist(self, alpha: TensorLike, *args, **kwargs):
return tfd.Zipf(power=alpha, dtype=tf.float32, *args, **kwargs)
