Python onehotの例

コード例 #1

ファイル: test_preprocessing.py プロジェクト: mylearning2017/deep-learning-book

 def test_dtype(self):
     oh_ary = onehot(ary=np.array([0, 1, 2]), dtype=np.int32)
     expect = np.array([[1, 0, 0],
                        [0, 1, 0],
                        [0, 0, 1]])
     self.assertTrue(np.array_equal(oh_ary, expect))
     self.assertTrue(oh_ary.dtype == np.int32)

コード例 #2

ファイル: test_preprocessing.py プロジェクト: mylearning2017/deep-learning-book

 def test_skiplabel(self):
     oh_ary = onehot(ary=np.array([0, 1, 2, 3, 5]))
     expect = np.array([[1., 0., 0., 0., 0., 0.],
                        [0., 1., 0., 0., 0., 0.],
                        [0., 0., 1., 0., 0., 0.],
                        [0., 0., 0., 1., 0., 0.],
                        [0., 0., 0., 0., 0., 1.]])
     self.assertTrue(np.array_equal(oh_ary, expect))

コード例 #3

ファイル: test_preprocessing.py プロジェクト: mylearning2017/deep-learning-book

 def test_defaults(self):
     oh_ary = onehot(ary=np.array([0, 1, 2, 3]))
     expect = np.array([[1., 0., 0., 0.],
                        [0., 1., 0., 0.],
                        [0., 0., 1., 0.],
                        [0., 0., 0., 1.]])
     self.assertTrue(np.array_equal(oh_ary, expect))
     self.assertTrue(oh_ary.dtype == np.float32)

コード例 #4

ファイル: test_preprocessing.py プロジェクト: zorro2017zdc/deep-learning-book

 def test_dtype(self):
     oh_ary = onehot(ary=np.array([0, 1, 2]), dtype=np.int32)
     expect = np.array([[1, 0, 0], [0, 1, 0], [0, 0, 1]])
     self.assertTrue(np.array_equal(oh_ary, expect))
     self.assertTrue(oh_ary.dtype == np.int32)

コード例 #5

ファイル: test_preprocessing.py プロジェクト: zorro2017zdc/deep-learning-book

 def test_n_classes(self):
     oh_ary = onehot(ary=np.array([0, 1, 2]), n_classes=5)
     expect = np.array([[1., 0., 0., 0., 0.], [0., 1., 0., 0., 0.],
                        [0., 0., 1., 0., 0.]])
     self.assertTrue(np.array_equal(oh_ary, expect))

コード例 #6

ファイル: test_preprocessing.py プロジェクト: zorro2017zdc/deep-learning-book

 def test_skiplabel(self):
     oh_ary = onehot(ary=np.array([0, 1, 2, 3, 5]))
     expect = np.array([[1., 0., 0., 0., 0., 0.], [0., 1., 0., 0., 0., 0.],
                        [0., 0., 1., 0., 0., 0.], [0., 0., 0., 1., 0., 0.],
                        [0., 0., 0., 0., 0., 1.]])
     self.assertTrue(np.array_equal(oh_ary, expect))

コード例 #7

ファイル: test_preprocessing.py プロジェクト: zorro2017zdc/deep-learning-book

 def test_defaults(self):
     oh_ary = onehot(ary=np.array([0, 1, 2, 3]))
     expect = np.array([[1., 0., 0., 0.], [0., 1., 0., 0.],
                        [0., 0., 1., 0.], [0., 0., 0., 1.]])
     self.assertTrue(np.array_equal(oh_ary, expect))
     self.assertTrue(oh_ary.dtype == np.float32)

コード例 #8

ファイル: test_preprocessing.py プロジェクト: mylearning2017/deep-learning-book

 def test_n_classes(self):
     oh_ary = onehot(ary=np.array([0, 1, 2]), n_classes=5)
     expect = np.array([[1., 0., 0., 0., 0.],
                        [0., 1., 0., 0., 0.],
                        [0., 0., 1., 0., 0.]])
     self.assertTrue(np.array_equal(oh_ary, expect))