Module: tfr.extension.pipeline
Stay organized with collections
Save and categorize content based on your preferences.
Provides a RankingPipeline for running a TF-Ranking model.
This class contains the boilerplate required to run a TF-Ranking model, which
reduces a few replicated setups (e.g., input function, serving input function,
model export strategies) for running TF-Ranking models. Advanced users can also
derive from this class and further tailor for their needs.
Classes
class RankingPipeline: Class to set up the input, train and eval processes for a TF Ranking model.
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.
Last updated 2023-08-18 UTC.
[null,null,["Last updated 2023-08-18 UTC."],[],[],null,["# Module: tfr.extension.pipeline\n\n\u003cbr /\u003e\n\n|---------------------------------------------------------------------------------------------------------------------|\n| [View source on GitHub](https://github.com/tensorflow/ranking/blob/v0.5.3/tensorflow_ranking/extension/pipeline.py) |\n\nProvides a `RankingPipeline` for running a TF-Ranking model.\n\nThis class contains the boilerplate required to run a TF-Ranking model, which\nreduces a few replicated setups (e.g., input function, serving input function,\nmodel export strategies) for running TF-Ranking models. Advanced users can also\nderive from this class and further tailor for their needs.\n\nClasses\n-------\n\n[`class RankingPipeline`](../../tfr/extension/pipeline/RankingPipeline): Class to set up the input, train and eval processes for a TF Ranking model."]]
View source on GitHub