PyTorch-Optimized: Cross-Entropy Loss

lgragert / nn-sero-pytorch

PyTorch version of neural network HLA serology prediction

2 stars 1 forks source link

Closed gbiagini closed 3 years ago

gbiagini commented 3 years ago

Currently using BCEWithLogits loss function (binary cross entropy) for multi-label multiclass classification.