Create ExampleProto's containing data.
(num_examples, input_mean)
| 37 | |
| 38 | |
| 39 | def create_examples(num_examples, input_mean): |
| 40 | """Create ExampleProto's containing data.""" |
| 41 | ids = np.arange(num_examples).reshape([num_examples, 1]) |
| 42 | inputs = np.random.randn(num_examples, 1) + input_mean |
| 43 | target = inputs - input_mean |
| 44 | examples = [] |
| 45 | for row in range(num_examples): |
| 46 | ex = example_pb2.Example() |
| 47 | ex.features.feature["id"].bytes_list.value.append(str(ids[row, 0])) |
| 48 | ex.features.feature["target"].float_list.value.append(target[row, 0]) |
| 49 | ex.features.feature["inputs"].float_list.value.append(inputs[row, 0]) |
| 50 | examples.append(ex) |
| 51 | return examples |
| 52 | |
| 53 | |
| 54 | def create_dir_test(): |