Python align_minibatches示例

示例#1

文件： test_minibatches.py 项目： yzh211/pymc3

 def test_align(self):
     m = pm.Minibatch(np.arange(1000), 1, random_seed=1)
     n = pm.Minibatch(np.arange(1000), 1, random_seed=1)
     f = theano.function([], [m, n])
     n.eval()  # not aligned
     a, b = zip(*(f() for _ in range(1000)))
     assert a != b
     pm.align_minibatches()
     a, b = zip(*(f() for _ in range(1000)))
     assert a == b
     n.eval()  # not aligned
     pm.align_minibatches([m])
     a, b = zip(*(f() for _ in range(1000)))
     assert a != b
     pm.align_minibatches([m, n])
     a, b = zip(*(f() for _ in range(1000)))
     assert a == b

示例#2

文件： test_minibatches.py 项目： alexander-belikov/pymc3

 def test_align(self):
     m = pm.Minibatch(np.arange(1000), 1, random_seed=1)
     n = pm.Minibatch(np.arange(1000), 1, random_seed=1)
     f = theano.function([], [m, n])
     n.eval()  # not aligned
     a, b = zip(*(f() for _ in range(1000)))
     assert a != b
     pm.align_minibatches()
     a, b = zip(*(f() for _ in range(1000)))
     assert a == b
     n.eval()  # not aligned
     pm.align_minibatches([m])
     a, b = zip(*(f() for _ in range(1000)))
     assert a != b
     pm.align_minibatches([m, n])
     a, b = zip(*(f() for _ in range(1000)))
     assert a == b