def test_output_shape(self):
# settings
batsize = 10
seqlen = 5
invocsize = 50
inembdim = 50
outembdim = 40
innerdim = 11
outvocsize = 17
# data
traindata = np.random.randint(0, invocsize, (batsize, seqlen))
traingold = np.random.randint(0, outvocsize, (batsize, seqlen))
# model
m = SimpleSeqTransDec(indim=invocsize, inpembdim=inembdim, outpembdim=outembdim, innerdim=innerdim, outdim=outvocsize)
pred = m.predict(traindata, shiftdata(traingold))
self.assertEqual(pred.shape, (batsize, seqlen, outvocsize))
def test_output_shape(self):
# settings
batsize = 10
seqlen = 5
invocsize = 50
inembdim = 50
outembdim = 40
innerdim = 11
outvocsize = 17
# data
traindata = np.random.randint(0, invocsize, (batsize, seqlen))
traingold = np.random.randint(0, outvocsize, (batsize, seqlen))
# model
m = SimpleSeqTransDec(
indim=invocsize, inpembdim=inembdim, outpembdim=outembdim, innerdim=innerdim, outdim=outvocsize
)
pred = m.predict(traindata, shiftdata(traingold))
self.assertEqual(pred.shape, (batsize, seqlen, outvocsize))
