code: def log_reg(
      input_shape: ShapeTuple, output_n: int, activation=None
  ) -> Model:
      inputs = klayers.Input(shape=input_shape)
      x = klayers.Flatten()(inputs)
      outputs = klayers.Dropout(0.25)(x, training=False)
      return kmodels.Model(
          inputs=inputs, outputs=outputs, name="log_reg"
      )
  
file_path: innvestigate/tests/networks/base.py
imports: from __future__ import annotations
  import tensorflow.keras.layers
  import tensorflow.keras.models
  from innvestigate.backend.types import Layer, Model, ShapeTuple
label_desc: Use image node_id 3 for calling a global function with example usage: log_reg(input_shape, output_n, activation) and returns: kmodels