Closed LeoVict closed 2 years ago
Hmm, it should be the paper's version. Could you give more detail? What is the command you are running? And what is you pytorch version?
I just downloaded the trained parameters and ran the "evaluate" code. ped2 AUC is just 86.2%
Hmmm weird.. Just now I tested on other system. Fresh with new git clone, downloaded the dataset and weight from link in the README. And still got 98.4% AUC. What is your pytorch version? I once saw other repo has problem with AUC number because of the pytorch version.
Hello, I used your parameter test and did not get the effect proposed in the paper. Is it because the code has not been updated