def train_valid_dict_generator(train_list, valid_list, df):
    train_df = {}
    for i in train_list:
        for j in i:
            lbl = df[(df['Image'] == j)]['Id'].values[0]
            train_df[j] = lbl
    valid_df = {}
    for i in valid_list:
        for j in i:
            lbl = df[(df['Image'] == j)]['Id'].values[0]
            valid_df[j] = lbl
    return (train_df, valid_df)
def train_valid_dict_generator(train_list, valid_list, df):
    train_df = {}
    for i in train_list:
        for j in i:
            lbl = df[(df['Image'] == j)]['Id'].values[0]
            train_df[j] = lbl
    valid_df = {}
    for i in valid_list:
        for j in i:
            lbl = df[(df['Image'] == j)]['Id'].values[0]
            valid_df[j] = lbl
    return (train_df, valid_df)