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)