Can this model be applied to predicting affinity value?

[agave233]

Hi,
Thanks for your solid work!

I noticed that this model can predict whether the input protein-ligand is active or not, which belongs to a binary classification task. However, I wonder GNN-DTI can be used for regression tasks? Can this model predict the binding affinity values well?

Thanks