def match()

in retail/recommendation-system/bqml-scann/tfx_pipeline/item_matcher.py [0:0]

• 8 lines of code
• 2 McCabe index (conditional complexity)

  def match(self, vector, num_matches=10):
    embedding = np.array(vector)
    query = embedding / np.linalg.norm(embedding)
    similarities = np.dot(self.embeddings, query.T) 
    matches = list(zip(self.tokens, list(similarities)))
    matches = sorted(
      matches, key=lambda kv: kv[1], reverse=True)[:num_matches]
    return [kv[0] for kv in matches]