Python iterate_minibatches Examples

Example #1

File: test_training.py Project: zqfang/deep-learning-book

 def test_default_maxbatch(self):
     a = np.array([1, 2, 3, 4, 5, 6])
     b = np.array([7, 8, 9, 10, 11, 12])
     it = iterate_minibatches((a, b), batch_size=6)
     m = list(next(it))
     self.assertTrue(np.array_equal(m[0], a))
     self.assertTrue(np.array_equal(m[1], b))

Example #2

File: test_training.py Project: mylearning2017/deep-learning-book

 def test_default_maxbatch(self):
     a = np.array([1, 2, 3, 4, 5, 6])
     b = np.array([7, 8, 9, 10, 11, 12])
     it = iterate_minibatches((a, b), batch_size=6)
     m = list(next(it))
     self.assertTrue(np.array_equal(m[0], a))
     self.assertTrue(np.array_equal(m[1], b))

Example #3

File: test_training.py Project: zqfang/deep-learning-book

 def test_default_uneven(self):
     a = np.array([1, 2, 3, 4, 5, 6])
     b = np.array([7, 8, 9, 10, 11, 12])
     expect1 = np.array([[1, 2, 3, 4]])
     expect2 = np.array([[7, 8, 9, 10]])
     for idx, m in enumerate(iterate_minibatches((a, b), batch_size=4)):
         m = list(m)
         self.assertTrue(np.array_equal(m[0], expect1[idx]))
         self.assertTrue(np.array_equal(m[1], expect2[idx]))

Example #4

File: test_training.py Project: mylearning2017/deep-learning-book

 def test_default_uneven(self):
     a = np.array([1, 2, 3, 4, 5, 6])
     b = np.array([7, 8, 9, 10, 11, 12])
     expect1 = np.array([[1, 2, 3, 4]])
     expect2 = np.array([[7, 8, 9, 10]])
     for idx, m in enumerate(iterate_minibatches((a, b), batch_size=4)):
         m = list(m)
         self.assertTrue(np.array_equal(m[0], expect1[idx]))
         self.assertTrue(np.array_equal(m[1], expect2[idx]))

Example #5

File: test_training.py Project: mylearning2017/deep-learning-book

 def test_shuffle(self):
     a = np.array([1, 2, 3, 4, 5, 6])
     b = np.array([7, 8, 9, 10, 11, 12])
     expect1 = np.array([[2, 4], [5, 1], [3, 6]])
     expect2 = np.array([[8, 10], [11, 7], [9, 12]])
     for idx, m in enumerate(iterate_minibatches((a, b), batch_size=2,
                             shuffle=True, seed=123)):
         m = list(m)
         self.assertTrue(np.array_equal(m[0], expect1[idx]))
         self.assertTrue(np.array_equal(m[1], expect2[idx]))

Example #6

File: test_training.py Project: zqfang/deep-learning-book

 def test_shuffle(self):
     a = np.array([1, 2, 3, 4, 5, 6])
     b = np.array([7, 8, 9, 10, 11, 12])
     expect1 = np.array([[2, 4], [5, 1], [3, 6]])
     expect2 = np.array([[8, 10], [11, 7], [9, 12]])
     for idx, m in enumerate(
             iterate_minibatches((a, b),
                                 batch_size=2,
                                 shuffle=True,
                                 seed=123)):
         m = list(m)
         self.assertTrue(np.array_equal(m[0], expect1[idx]))
         self.assertTrue(np.array_equal(m[1], expect2[idx]))

Example #7

File: test_training.py Project: zqfang/deep-learning-book

 def test_default_hugebatch(self):
     a = np.array([1, 2, 3, 4, 5, 6])
     b = np.array([7, 8, 9, 10, 11, 12])
     it = iterate_minibatches((a, b), batch_size=99)
     with self.assertRaises(StopIteration):
         m = next(it)

Example #8

File: test_training.py Project: mylearning2017/deep-learning-book

 def test_default_hugebatch(self):
     a = np.array([1, 2, 3, 4, 5, 6])
     b = np.array([7, 8, 9, 10, 11, 12])
     it = iterate_minibatches((a, b), batch_size=99)
     with self.assertRaises(StopIteration):
         m = next(it)