Graphormer icon indicating copy to clipboard operation
Graphormer copied to clipboard

How to evaluate the model with V.2?

Open AmeenAli opened this issue 3 years ago • 5 comments

Hi, How can I print the eval accuracy after each epoch in the new version of Graphormer?

AmeenAli avatar Dec 23 '21 12:12 AmeenAli

graphormer/criterions/binary_logloss.py provides the accuracy metrics, and if your objective is a multi-task binary classification, it will count the average accuracy over all tasks. If you want to use other metric, like AUC, you can refer to this.

zhengsx avatar Dec 23 '21 12:12 zhengsx

Thanks! where does this gets called? as i dont see from out output logging the acc of the validation/test

AmeenAli avatar Dec 23 '21 12:12 AmeenAli

do you use --criterion binary_logloss ?

zhengsx avatar Dec 23 '21 12:12 zhengsx

I am running the example of ZIN where --criterion l1_loss , and it still does not output validation.

AmeenAli avatar Dec 23 '21 12:12 AmeenAli

With l1_loss, the loss is exactly the MAE. And the loss of valid set should be evaluated after every epoch. Running zinc.sh script in the example folder gives following log log

shiyu1994 avatar Dec 23 '21 12:12 shiyu1994