code: def train(feature_data, labels):
      data = feature_data
      labels = labels
  
      db_train = gen_tensorflow_dataset((data, labels))
      assert isinstance(db_train, DatasetV2)
      db_train = db_train.batch(32)
  
      model = get_model()
      model.fit(db_train, epochs=2)
  
file_path: mars/mars/learn/contrib/tensorflow/tests/tf_dataset.py
imports: import os
  import json
  import sys
  import tensorflow
  from tensorflow.keras import layers
  from mars.learn.contrib.tensorflow import gen_tensorflow_dataset
  from tensorflow.python.data.ops.dataset_ops import DatasetV2
label_desc: Use image node_id 2 for calling a global function with example usage: train(feature_data, labels) without return types