RayJue/retrieve_and_rank

This repository implements a movie recommendation system using TensorFlow Recommenders (TFRS) with the MovieLens 100K dataset. It features retrieval and ranking models employing a two-tower architecture for user and movie embeddings, and demonstrates a brute-force search method for candidate retrieval.

Python