def __call__()

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

• 10 lines of code
• 1 McCabe index (conditional complexity)

  def __call__(self, inputs):
    tokens = tf.strings.split(inputs, sep=None).to_sparse()
    ids = self.token_to_id.lookup(tokens) 
    embeddings = tf.nn.embedding_lookup_sparse(
        params=self.embeddings, 
        sp_ids=ids, 
        sp_weights=None, 
        combiner="mean"
    )
    return embeddings