ML4SCI / DeepLearnHackathon

13 stars 9 forks source link

Fixed nums not defined. Increased batchsize, cleared some cell outputs #33

Closed BoGGoG closed 3 months ago

BoGGoG commented 3 months ago

nums was only defined in the cell after, so I moved it up one cell.

If batchsize is too small, there will only be one class in the batch and roc_auc_score failed on validation epoch end.