-
Notifications
You must be signed in to change notification settings - Fork 0
Description
Hello again :)
Having tried different approach to make small changes to DivHGNN, I wanted to have an idea of exactly how each part of your design contributes to the very good result highlithed in the paper. Nonetheless, I cannot seem to reproduce it with the simple full_eval() ran every epoch. Without pruning, I only get up to 66.45% AUC when running full_eval(). The paper mentions on average convergence after 24 epochs, how exactly do you choose the model? By simply taking the best performing one (ie no val dataset)? Full_eval does the exponential decay, but I noticed that if I run full_eval on a fraction of the edges, say the first 10'000 sessions, I get much better result than running on everything as well, is that expected?
I'm sorry for digging up into the model, but I greatly appreciate what you have built and I'd love to be able to accurately reproduce it.