code: def _train_model(model):
      loss_fn = tf.keras.losses.SparseCategoricalCrossentropy(
          from_logits=True
      )
  
      model.compile(
          optimizer="adam", loss=loss_fn, metrics=["accuracy"]
      )
  
      (x_train, y_train), _ = _get_dataset()
  
      model.fit(x_train, y_train, epochs=EPOCHS)
  
imports: import tensorflow
  from tensorflow_model_optimization.python.core.clustering.keras import cluster
  from tensorflow_model_optimization.python.core.clustering.keras import cluster_config
  from tensorflow_model_optimization.python.core.clustering.keras import clusterable_layer
  from tensorflow_model_optimization.python.core.clustering.keras import clustering_algorithm
label_desc: Use image node_id 4 for calling a global function with example usage: _train_model(model) without return types