def test_one_hot_batches_invalid_input(self):
wrapper = OneHotEncoding(
DataStream(IndexableDataset(self.data),
iteration_scheme=SequentialScheme(4, 2)),
num_classes=2,
which_sources=('targets',))
assert_raises(ValueError, list, wrapper.get_epoch_iterator())
def test_one_hot_batches_invalid_input(self):
wrapper = OneHotEncoding(DataStream(IndexableDataset(self.data),
iteration_scheme=SequentialScheme(
4, 2)),
num_classes=2,
which_sources=('targets', ))
assert_raises(ValueError, list, wrapper.get_epoch_iterator())
def test_one_hot_examples(self):
wrapper = OneHotEncoding(DataStream(
IndexableDataset(self.data),
iteration_scheme=SequentialExampleScheme(4)),
num_classes=4,
which_sources=('targets', ))
assert_equal(list(wrapper.get_epoch_iterator()),
[(numpy.ones((2, 2)), numpy.array([[1, 0, 0, 0]])),
(numpy.ones((2, 2)), numpy.array([[0, 1, 0, 0]])),
(numpy.ones((2, 2)), numpy.array([[0, 0, 1, 0]])),
(numpy.ones((2, 2)), numpy.array([[0, 0, 0, 1]]))])
def test_one_hot_batches(self):
wrapper = OneHotEncoding(
DataStream(IndexableDataset(self.data),
iteration_scheme=SequentialScheme(4, 2)),
num_classes=4,
which_sources=('targets',))
assert_equal(
list(wrapper.get_epoch_iterator()),
[(numpy.ones((2, 2, 2)),
numpy.array([[1, 0, 0, 0], [0, 1, 0, 0]])),
(numpy.ones((2, 2, 2)),
numpy.array([[0, 0, 1, 0], [0, 0, 0, 1]]))])
