class_docstr: Trains Keras models.
class_docstr_tok: ["Trains","Keras","models","."]
code: def get_train_dataset(self) -> tf.data.Dataset:
      return self._train_dataset
  
file_path: adanet/adanet/experimental/phases/keras_trainer_phase.py
imports: from typing import Callable, Iterable, Iterator, Union
  from adanet.experimental.phases.phase import DatasetProvider
  from adanet.experimental.phases.phase import ModelProvider
  from adanet.experimental.storages.in_memory_storage import InMemoryStorage
  from adanet.experimental.storages.storage import Storage
  from adanet.experimental.work_units.keras_trainer_work_unit import KerasTrainerWorkUnit
  from adanet.experimental.work_units.work_unit import WorkUnit
  import tensorflow.compat.v2
label_desc: Use image node_id 5 for calling the KerasTrainerPhase obj's underlying member method code with example usage: obj.get_train_dataset() and returns: self