4 ms·
Does PyTorch have a learn to rank module? Tensorflow released a ranking module earlier this year, but I’d like to try out PyTorch.
by BillFranklin 7y ago
Does PyTorch have a learn to rank module? Tensorflow released a ranking module earlier this year, but I’d like to try out PyTorch.
- geraltofrivia 7y agoNot as far I know. It does have max-margin loss [1], which is pretty much all you need to implement a neural ranking model, apart from data iterators, and training loops. [1] (https://pytorch.org/docs/stable/nn.html?highlight=margin%20loss#torch.nn.MarginRankingLoss https://pytorch.org/docs/stable/nn.html?highlight=margin%20l...)