anonym844 / KG-diverse

5 stars 0 forks source link

A problem about reproduction #2

Open DDCY220 opened 5 months ago

DDCY220 commented 5 months ago

Hello, can you provide the optimal parameters of the three data sets for reproduction?

Xiaolong-Liu-bdsc commented 5 months ago

Hello, can you provide the optimal parameters of the three data sets for reproduction?

Thanks for your interest in our work. Due to the limitations in time and computational resources, we focused on a set of parameters that we identified as optimal within our experimental framework. Here are the parameters we used: lr / wd / embed_size / n_layers / KG_layers / lambda_1 (gamma) / lambda_2 (beta) Last.fm 0.01 / 1e-05 / 64 / 1 / 5 / 0.9 / 0.45 Movie_lens 0.005 / 1e-07 / 64 / 4 / 1 / 0.1 / 0.1 Amazon-Book 0.05 / 0 / 64 / 3 / 4 / 0.3 / 0.15