提交 546cc762 authored 作者: Frederic's avatar Frederic

pep8

上级 f585ac8b
......@@ -281,29 +281,30 @@ def test_trace():
ok = True
assert ok
def test_spectral_radius_bound():
tol = 10**(-6)
tol = 10 ** (-6)
rng = numpy.random.RandomState(utt.fetch_seed())
x = theano.tensor.matrix()
radius_bound = spectral_radius_bound(x, 5)
f = theano.function([x], radius_bound)
shp = (3, 3)
shp = (3, 4)
m = rng.rand(*shp).astype(config.floatX)
m = numpy.cov(m).astype(config.floatX)
radius_bound_theano = f(m)
# test the approximation
mm = m
for i in range(5):
mm = numpy.dot(mm, mm)
radius_bound_numpy = numpy.trace(mm)**(2**(-5))
radius_bound_numpy = numpy.trace(mm) ** (2 ** (-5))
assert abs(radius_bound_numpy - radius_bound_theano) < tol
# test the bound
# test the bound
eigen_val = numpy.linalg.eig(m)
assert (eigen_val[0].max() - radius_bound_theano) < tol
# test type errors
xx = theano.tensor.vector()
ok = False
......@@ -318,7 +319,7 @@ def test_spectral_radius_bound():
except TypeError:
ok = True
assert ok
# test value error
ok = False
try:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论