def train_input_fn():
    input_dict = {'input_a': train_data['input_a'], 'input_b': train_data['input_b'], 'input_m': train_data['input_m'].astype(np.str)}
    output_dict = {'dense_2': train_data['output_c'], 'dense_3': train_data['output_d']}
    return dataset_ops.Dataset.from_tensor_slices((input_dict, output_dict)).batch(16)
def train_input_fn():
    input_dict = {'input_a': train_data['input_a'], 'input_b': train_data['input_b'], 'input_m': train_data['input_m'].astype(np.str)}
    output_dict = {'dense_2': train_data['output_c'], 'dense_3': train_data['output_d']}
    return dataset_ops.Dataset.from_tensor_slices((input_dict, output_dict)).batch(16)