Wonbin Kweon, SeongKu Kang, and Hwanjo Yu. 2022. Obtaining Calibrated Probabilities with Personalized Ranking Models. Thirty-Sixth AAAI Conference on Artificial Intelligence (AAAI oral, 3% accepted). \
This is a Pytorch implementation.
Version:
pytorch 1.7.0 \
-
we cannot include the pre-processed dataset due to the lack of space. Instead, you can run yahooR3.ipynb in /data to pre-process the raw dataset.
-
Please install all the libraries used in main.py
-
run main.py