sdf.py

def knearest(r_id, set_of_restaurants, db, k = 7, reg = 3.):
    def getTupel(r):
        sim, support = db.get(r_id, r)
        shrunken_sim = shrunk_sim(sim, support, reg)
        return (r, shrunken_sim, support)
    rs = [getTupel(r) for r in set_of_restaurants]
    return sorted(rs, key=lambda r: -r[1])[:k]