| 157 | |
| 158 | |
| 159 | def build_network_payload(layers: Sequence[LayerMetadata]) -> dict[str, Any]: |
| 160 | if not layers: |
| 161 | raise ValueError("Layer metadata must contain at least one dense layer.") |
| 162 | architecture = [layers[0].weight_shape[1]] + [layer.bias_shape[0] for layer in layers] |
| 163 | return { |
| 164 | "architecture": architecture, |
| 165 | "layers": [ |
| 166 | { |
| 167 | "layer_index": layer.layer_index, |
| 168 | "type": layer.type, |
| 169 | "name": layer.name, |
| 170 | "activation": layer.activation, |
| 171 | "weight_shape": list(layer.weight_shape), |
| 172 | "bias_shape": list(layer.bias_shape), |
| 173 | } |
| 174 | for layer in layers |
| 175 | ], |
| 176 | "input_dim": layers[0].weight_shape[1], |
| 177 | "output_dim": layers[-1].bias_shape[0], |
| 178 | "normalization": {"mean": MNIST_MEAN, "std": MNIST_STD}, |
| 179 | } |
| 180 | |
| 181 | |
| 182 | def slugify_identifier(value: str) -> str: |