loaded scann index can not use k param

https://github.com/tensorflow/recommenders/blob/main/docs/examples/basic_retrieval.ipynb

# Export the query model.
with tempfile.TemporaryDirectory() as tmp:
  path = os.path.join(tmp, "model")

  # Save the index.
  index.save(path)

  # Load it back; can also be done in TensorFlow Serving.
  loaded = tf.keras.models.load_model(path)

  # Pass a user id in, get top predicted movie titles back.
  scores, titles = loaded(["42"])

  print(f"Recommendations: {titles[0][:3]}")

loaded object have not k param,so i can not return k(default 10) than 10 movies?

[alimirferdos]

@maciejkula I have two issues concerning the answer you gave above.

First, although I call the functions before creating concrete functions, it still requires me to pass input specs. I got some results using the following snippet but I don’t think it’s the right way to do so:

recom_10 = tf.function(lambda user_id: model(user_id, k=10))
_ = recom_10(tf.constant([812]))

signatures = {"k_10": recom_10.get_concrete_function(user_id=tf.constant([812]))}

My second issue was creating a concrete function for query_with_exclusions. I’ve used the following code:

excluded_recoms = tf.function(lambda user_id, exclusions:
                              index.query_with_exclusions(user_id, k=10, exclusions=exclusions))
_ = excluded_recoms(uid=tf.constant([812]), exclusions=tf.constant([[150]]))

signatures = {"excluded_recoms": excluded_recoms.get_concrete_function(uid=tf.constant([812]), exclusions=tf.constant([[150]]))}

This way I can save and load the index but I can only pass a single exclusion id because its shape is set to (1, 1).

Any idea on how to solve these two issues?

[sam-watts]

@maciejkula is there a reason why (other than it feeling like a bit of a hack…) you couldn’t just define the argument k in topK.query_with_exclusions as a tf.Tensor and then just extract the value from that? Would that allow you to define it via TensorSpec to get a concrete function that allows for multiple values of k at inference time?