in src/utils/recipe1m_utils.py [0:0]
def cluster_ingredients(counter_ingrs):
mydict = dict()
mydict_ingrs = dict()
for k, v in counter_ingrs.items():
w1 = k.split('_')[-1]
w2 = k.split('_')[0]
lw = [w1, w2]
if len(k.split('_')) > 1:
w3 = k.split('_')[0] + '_' + k.split('_')[1]
w4 = k.split('_')[-2] + '_' + k.split('_')[-1]
lw = [w1, w2, w4, w3]
gotit = 0
for w in lw:
if w in counter_ingrs.keys():
# check if its parts are
parts = w.split('_')
if len(parts) > 0:
if parts[0] in counter_ingrs.keys():
w = parts[0]
elif parts[1] in counter_ingrs.keys():
w = parts[1]
if w in mydict.keys():
mydict[w] += v
mydict_ingrs[w].append(k)
else:
mydict[w] = v
mydict_ingrs[w] = [k]
gotit = 1
break
if gotit == 0:
mydict[k] = v
mydict_ingrs[k] = [k]
return mydict, mydict_ingrs