DeepRank / Deeprank-GNN

Graph Network for protein-protein interface
Apache License 2.0
117 stars 32 forks source link

update classification output format #60

Closed manonreau closed 3 years ago

manonreau commented 3 years ago

the Cross Entropy loss function takes as input the raw scores for each class. To meet this requirement, the softmax activation function is applied a posteriori to get save a probability of belonging to each class.