コード例 #1
0
 def _base_dist(self, p: TensorLike, *args, **kwargs):
     return tfd.Geometric(probs=p, *args, **kwargs)
コード例 #2
0
ファイル: discrete.py プロジェクト: saforem2/pymc4 
 def _init_distribution(conditions):
     p = conditions["p"]
     return tfd.Geometric(probs=p)
コード例 #3
0
ファイル: discrete.py プロジェクト: christhorn2/pymc4 
 def _init_distribution(conditions, **kwargs):
     probs = conditions["probs"]
     return tfd.Geometric(probs=probs, **kwargs)