Vlad Lujerdeanu
Vlad Lujerdeanu
Never mind, I think I got it in the end. If anyone needs it: ``` class RankingModel(tfrs.Model): def __init__(self, query_model, candidate_model): super().__init__() self.query_model = query_model self.candidate_model = candidate_model self.ratings =...
Ok. Apparently I was wrong (I think). The models works fine, but only if it uses the same candidate tower for retrieval and ranking, and it needs to be trained...
I didn't, I just used the same candidate and query towers from retrieval in the ranking phase. It worked but I don't think it's how it should be solved.