Python create_random_dataset 예제들

예제 #1

파일: tapas_pretraining_model_test.py 프로젝트: SOI-TechNinjas/Tapas-Demo

 def _predict_input_fn(params):
     dataset = table_dataset_test_utils.create_random_dataset(
         num_examples=params["batch_size"],
         batch_size=params["batch_size"],
         repeat=True,
         generator_kwargs=generator_kwargs)
     return dataset.take(2)

예제 #2

파일: prediction_utils_test.py 프로젝트: wcc17864158993/tapas

 def _input_fn(params):
     return table_dataset_test_utils.create_random_dataset(
         num_examples=params['batch_size'] * 2,
         batch_size=params['batch_size'],
         repeat=False,
         generator_kwargs=self._generator_kwargs(
             add_aggregation_function_id=do_model_aggregation,
             add_classification_labels=do_model_classification,
         ))

예제 #3

파일: prediction_utils_test.py 프로젝트: wcc17864158993/tapas

 def _input_fn(params):
     return table_dataset_test_utils.create_random_dataset(
         num_examples=params['batch_size'] * 2,
         batch_size=params['batch_size'],
         repeat=False,
         generator_kwargs=self._generator_kwargs())

예제 #4

 def _input_fn(params):
     return table_dataset_test_utils.create_random_dataset(
         num_examples=params["batch_size"],
         batch_size=params["batch_size"],
         repeat=False,
         generator_kwargs=generator_kwargs)