Hi
I ran the below command (had to fetch the pretrained model from the dMaSIF_colab repo) to test the site prediction accuracy and the ROC-AUC was on average of 0.62, not the 0.87 you state in the paper. How did you get this figure?
python --experiment_name dMaSIF_site_3layer_16dims_9A_100sup_epoch64 --batch_size 64 --embedding_layer dMaSIF --site True --emb_dims 16 --device cuda:0 --radius 9.0 --n_layers 3
I also ran the same command but for search prediction and I got an average ROC-AUC of ~0.5 (so essentially random).
It would be really helpful if you could explain how to use this tool to get the results you say you obtained in the paper.
Thanks