amzn / pecos

PECOS - Prediction for Enormous and Correlated Spaces
https://libpecos.org/
Apache License 2.0
517 stars 105 forks source link

Refactor RankingModel class for Text+Numr use case #299

Closed OctoberChang closed 2 months ago

OctoberChang commented 2 months ago

Issue #, if available:

Description of changes: Refactor the PECOS RankingModel class to support applications with multi-modal inputs, such as Text and Numerical Inputs.

Example Usage See pecos/examples/msmarco-rankllama/ for the text-only re-ranking applications.

By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.