Skip to content

Evaluating DivHGNN #2

@emduc

Description

@emduc

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.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions