def make_estimator_input_fn(self,
                              mode,
                              hparams,
                              data_dir=None,
                              force_repeat=False,
                              prevent_repeat=False,
                              dataset_kwargs=None):
    """Return input_fn wrapped for Estimator."""

    def estimator_input_fn(params, config):
      return self.input_fn(
          mode,
          hparams,
          data_dir=data_dir,
          params=params,
          config=config,
          force_repeat=force_repeat,
          prevent_repeat=prevent_repeat,
          dataset_kwargs=dataset_kwargs)

    return estimator_input_fn
def make_estimator_input_fn ( self , mode , hparams , data_dir = None , force_repeat = False , prevent_repeat = False , dataset_kwargs = None ) : def estimator_input_fn ( params , config ) : return self . input_fn ( mode , hparams , data_dir = data_dir , params = params , config = config , force_repeat = force_repeat , prevent_repeat = prevent_repeat , dataset_kwargs = dataset_kwargs ) return estimator_input_fn
Return input_fn wrapped for Estimator.