def test_tuple_conversion_from_tuple_datset(self):
x = np.random.rand(6, 1)
y = 2 * x + 3
tuple_dataset = tf.data.Dataset.from_tensor_slices((x, y))
converted_dataset = training_utils.convert_to_tuple_dataset(
tuple_dataset)
tuple_batch = next(iter(tuple_dataset))
converted_batch = next(iter(converted_dataset))
self.assertAllClose(tuple_batch, converted_batch)
def test_tuple_conversion_from_ordered_dict(self):
x = np.random.rand(6, 1)
y = 2 * x + 3
tuple_dataset = tf.data.Dataset.from_tensor_slices((x, y))
ordered_dict_dataset = tf.data.Dataset.from_tensor_slices(
collections.OrderedDict([('x', x), ('y', y)]))
converted_dataset = training_utils.convert_to_tuple_dataset(
ordered_dict_dataset)
tuple_batch = next(iter(tuple_dataset))
converted_batch = next(iter(converted_dataset))
self.assertAllClose(tuple_batch, converted_batch)
def test_tuple_conversion_from_named_tuple(self):
x = np.random.rand(6, 1)
y = 2 * x + 3
tuple_dataset = tf.data.Dataset.from_tensor_slices((x, y))
example_tuple = collections.namedtuple('Example', ['x', 'y'])
named_tuple_dataset = tf.data.Dataset.from_tensor_slices(
example_tuple(x=x, y=y))
converted_dataset = training_utils.convert_to_tuple_dataset(
named_tuple_dataset)
tuple_batch = next(iter(tuple_dataset))
converted_batch = next(iter(converted_dataset))
self.assertAllClose(tuple_batch, converted_batch)
