Skip to content

TensorFlow Ranking v0.3.2

Compare
Choose a tag to compare
@HongleiZhuang HongleiZhuang released this 19 Aug 21:58
· 343 commits to master since this release

In the latest release of TensorFlow Ranking v0.3.2, we introduce TFR-BERT extension to better support ranking models for text data based on BERT. BERT is a pre-trained language representation model which has achieved substantial improvement over numerous NLP tasks. We find that fine-tuning BERT with ranking losses further improve the ranking performance (arXiv). You can read detailed information about what is included in TFR-BERT extension here. There is also an example showing how to use TFR-BERT here.