code: def __init__(
      self,
      features_out: int,
      loss: tp.Any = None,
      metrics: tp.Any = None,
      optimizer=None,
      seed: int = 42,
      eager: bool = False,
  ):
      self.features_out = features_out
      super().__init__(
          module=None,
          loss=loss,
          metrics=metrics,
          optimizer=optimizer,
          seed=seed,
          eager=eager,
      )
  
file_path: elegy/examples/jax/linear_classifier_train_step.py
imports: import os
  import typing
  from datetime import datetime
  import jax
  import jax.numpy
  import numpy
  import optax
  import typer
  from datasets.load import load_dataset
  import elegy
label_desc: Use image node_id 1 to create a new Model object from inherited base classes: eg with example: obj = Model(features_out, loss, metrics, optimizer, seed, eager)