(dir_path='/tmp/mnist')
| 83 | |
| 84 | |
| 85 | def load(dir_path='/tmp/mnist'): |
| 86 | train_x, train_y, val_x, val_y = load_dataset(dir_path) |
| 87 | train_x, val_x = normalize(train_x, val_x) |
| 88 | train_x = train_x.astype(np.float32) |
| 89 | val_x = val_x.astype(np.float32) |
| 90 | train_y = train_y.astype(np.int32) |
| 91 | val_y = val_y.astype(np.int32) |
| 92 | return train_x, train_y, val_x, val_y |
nothing calls this directly
no test coverage detected