def awf_create_train_dataset_fn(model_input_config_file,
data_file,
batch_size,
file_type=None,
shuffle=False,
shuffle_buffer_size=None,
drop_remainder=False,
cache=False,
repeat=False,
**kwargs):
return create_dataset(model_input_config_file, data_file, file_type, None,
batch_size, shuffle, shuffle_buffer_size,
drop_remainder, cache, repeat, **kwargs)
def awf_create_predict_dataset_fn(model_input_config_file=None,
data_file=None,
batch_size=None,
file_type=None,
shuffle=False,
shuffle_buffer_size=None,
drop_remainder=False,
repeat=False,
**kwargs):
if not model_input_config_file or not data_file:
return None
return create_dataset(model_input_config_file, data_file, file_type, None,
batch_size, shuffle, shuffle_buffer_size,
drop_remainder, False, repeat, **kwargs)