search

lgragert / nn-sero-pytorch

PyTorch version of neural network HLA serology prediction
2 stars 1 forks source link

Rewrite NN using vanilla PyTorch #40

Open gbiagini opened 3 years ago

gbiagini commented 3 years ago

Rewrite NN using vanilla PyTorch rather than the Fast.Ai framework.

Send tasks to GPU as appropriate. One-hot encode serological predictions to ease use of multilabel classifier.