# Can i export retrieval model with query and candidate as inputs?

**URL:** <https://discuss.ai.google.dev/t/can-i-export-retrieval-model-with-query-and-candidate-as-inputs/27320>\
**Category:** General Discussion\
**Tags:** tfmodel, tf-dataset\
**Created:** [March 21, 2024, 4:53am UTC](https://discuss.ai.google.dev/t/can-i-export-retrieval-model-with-query-and-candidate-as-inputs/27320 "2024-03-21T04:53:45Z")\
**Posts on this page:** 1\
**Page:** 1

<div class="post-metadata">

**Author:** ![viniciuscheim](https://d1dlmcr85iqnpo.cloudfront.net/user_avatar/discuss.ai.google.dev/viniciuscheim/32/17209_2.png) [@viniciuscheim](https://discuss.ai.google.dev/u/viniciuscheim)\
**Post date:** [March 21, 2024, 4:53am UTC](https://discuss.ai.google.dev/t/can-i-export-retrieval-model-with-query-and-candidate-as-inputs/27320/1 "2024-03-21T04:53:45Z")

</div>

I have a factorized topK retrieval model, everything works fine, but I want to input specifics slices of candidates data, its possible? In the code i want to turn parsed\_topk dataset in to a input of the model

```auto
index = tfrs.layers.factorized_top_k.BruteForce(final_model.query_model)

index.index_from_dataset(
    tf.data.Dataset.zip((parsed_topK.batch(128).map(lambda x: x['experience.jobTitle']), parsed_topK.batch(128).map(final_model.candidate_model)))
)

```