提交 9d410ebf authored 作者: Ricardo's avatar Ricardo 提交者: ricardoV94

Add TriangularRV

上级 c3f2a8c3
...@@ -31,6 +31,17 @@ class UniformRV(RandomVariable): ...@@ -31,6 +31,17 @@ class UniformRV(RandomVariable):
uniform = UniformRV() uniform = UniformRV()
class TriangularRV(RandomVariable):
name = "triangular"
ndim_supp = 0
ndims_params = [0, 0, 0]
dtype = "floatX"
_print_name = ("Triang", "\\operatorname{Triang}")
triangular = TriangularRV()
class BetaRV(RandomVariable): class BetaRV(RandomVariable):
name = "beta" name = "beta"
ndim_supp = 0 ndim_supp = 0
......
...@@ -40,6 +40,7 @@ from aesara.tensor.random.basic import ( ...@@ -40,6 +40,7 @@ from aesara.tensor.random.basic import (
poisson, poisson,
polyagamma, polyagamma,
randint, randint,
triangular,
truncexpon, truncexpon,
uniform, uniform,
wald, wald,
...@@ -102,6 +103,15 @@ def test_uniform_samples(): ...@@ -102,6 +103,15 @@ def test_uniform_samples():
rv_numpy_tester(uniform, test_low, test_high, size=[3]) rv_numpy_tester(uniform, test_low, test_high, size=[3])
def test_triangular_samples():
test_left = np.array(10, dtype=config.floatX)
test_mode = np.array(12, dtype=config.floatX)
test_right = np.array(20, dtype=config.floatX)
rv_numpy_tester(triangular, test_left, test_mode, test_right)
rv_numpy_tester(triangular, test_left, test_mode, test_right, size=[2, 3])
def test_beta_samples(): def test_beta_samples():
test_a = np.array(0.5, dtype=config.floatX) test_a = np.array(0.5, dtype=config.floatX)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论