SML

Symmetric Metric Learning with Adaptive Margin for Recommendation (AAAI2020)

Running

python train.py --model SML --dataset aiv --hidden_size 100 --lr 0.1 --lamda 0.01 --gama 10

Requirements

Python3

Cuda 10

Tensorflow 1.4.0