Mock dataset function.
(feature_name)
| 68 | |
| 69 | |
| 70 | def mock_data(feature_name): |
| 71 | """Mock dataset function.""" |
| 72 | |
| 73 | def _generate_data(_): |
| 74 | x = tf.zeros(shape=(2,), dtype=tf.float32) |
| 75 | label = tf.zeros([1], dtype=tf.int32) |
| 76 | return {feature_name: x}, label |
| 77 | |
| 78 | dataset = tf.data.Dataset.range(1) |
| 79 | dataset = dataset.repeat() |
| 80 | dataset = dataset.map( |
| 81 | _generate_data, num_parallel_calls=tf.data.experimental.AUTOTUNE) |
| 82 | return dataset.prefetch(buffer_size=1).batch(2, drop_remainder=True) |
| 83 | |
| 84 | |
| 85 | class FooConfig(cfg.TaskConfig): |
no test coverage detected