anirudhacharya commented on a change in pull request #14617: PDF operators for
the random samplers, and also the Dirichlet
URL: https://github.com/apache/incubator-mxnet/pull/14617#discussion_r274654696
##########
File path: tests/python/unittest/test_random.py
##########
@@ -290,11 +315,84 @@ def check_with_device(device, dtype):
for check_name, check_func, tol in symbdic['checks']:
assert np.abs(check_func(samples, params)) < tol,
"symbolic test: %s check for `%s` did not pass" % (check_name, name)
-@with_seed()
+ # check pdfs with only a subset of the generated samples
+ un1 = np.resize(un1, (un1.shape[0], un1.shape[1], pdfshape[0],
pdfshape[1]))
+ print(name)
+ symbol = symbdic['pdfsymbol']
+ pdffunc = symbdic['pdffunc']
+ v0 = mx.sym.Variable('v0')
+ v1 = mx.sym.Variable('v1')
+ v2 = mx.sym.Variable('v2')
+ p1 = np.array(symbdic['inputs'][0][1])
+ p2 = None if single_param else np.array(symbdic['inputs'][1][1])
+ # Move samples away from boundaries of support
+ if name == 'gamma' or name == 'exponential':
+ un1 = np.maximum(un1, 1e-1)
+ if name == 'uniform':
+ un1 =
np.minimum(np.maximum(un1.reshape((un1.shape[0],un1.shape[1],-1)),
p1.reshape((p1.shape[0],p1.shape[1],-1))+1e-4),
+
p2.reshape((p2.shape[0],p2.shape[1],-1))-1e-4).reshape(un1.shape)
+ for use_log in [False, True]:
+ test_pdf = symbol(v0, v1, is_log=use_log) if single_param else
symbol(v0, v1, v2, is_log=use_log)
+ forw_atol = 1e-7 if dtype != np.float16 else 1e-3
+ forw_rtol = 1e-4 if dtype != np.float16 else 5e-2
+ backw_atol = 1e-3
+ backw_rtol = 5e-2
+ if single_param:
+ res = pdffunc(un1.reshape((un1.shape[0],un1.shape[1],-1)),
+
p1.reshape((p1.shape[0],p1.shape[1],-1))).reshape(un1.shape)
+ if use_log:
+ res = np.log(res)
+ check_symbolic_forward(test_pdf, [un1, p1], [res],
atol=forw_atol, rtol=forw_rtol, dtype=dtype)
+ if dtype == np.float64:
+ grad_nodes = ['v1'] if symbdic['discrete'] else ['v0', 'v1']
+ check_numeric_gradient(test_pdf, [un1, p1],
grad_nodes=grad_nodes, atol=backw_atol, rtol=backw_rtol, dtype=dtype)
+ else:
+ res = pdffunc(un1.reshape((un1.shape[0],un1.shape[1],-1)),
+ p1.reshape((p1.shape[0],p1.shape[1],-1)),
+
p2.reshape((p2.shape[0],p2.shape[1],-1))).reshape(un1.shape)
+ if use_log:
+ res = np.log(res)
+ check_symbolic_forward(test_pdf, [un1, p1, p2], [res],
atol=forw_atol, rtol=forw_rtol, dtype=dtype)
+ if dtype == np.float64:
+ grad_nodes = ['v1', 'v2'] if symbdic['discrete'] else ['v0',
'v1', 'v2']
+ check_numeric_gradient(test_pdf, [un1, p1, p2],
grad_nodes=grad_nodes, atol=backw_atol, rtol=backw_rtol, dtype=dtype)
+
+@with_seed(1000)
Review comment:
any reason why the seed has been fixed here?
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
With regards,
Apache Git Services
