code: def get_symbol_train(network, data_shape, **kwargs):
      
      if network.startswith('legacy'):
          logging.warn('Using legacy model.')
          return symbol_builder.import_module(network).get_symbol_train(**kwargs)
      config = get_config(network, data_shape, **kwargs).copy()
      config.update(kwargs)
      return symbol_builder.get_symbol_train(**config)
docs: Wrapper for get symbol for train
  
      Parameters
      ----------
      network : str
          name for the base network symbol
      data_shape : int
          input shape
      kwargs : dict
          see symbol_builder.get_symbol_train for more details