epoch    tot_tloss    gw_tloss    tot_vloss    gw_vloss    train_acc    valid_acc    roc_auc    
0    0.6941    0.6941    0.6935    0.6935    0.5074    0.505    0.5144    
1    0.6939    0.6939    0.693    0.693    0.5004    0.5075    0.5366    
2    0.6941    0.6941    0.6928    0.6928    0.4944    0.5125    0.5497    
3    0.6936    0.6936    0.6935    0.6935    0.5006    0.4835    0.6311    
4    0.6934    0.6934    0.6931    0.6931    0.5008    0.494    0.6503    
5    0.6936    0.6936    0.6925    0.6925    0.5028    0.4975    0.7283    
6    0.6924    0.6924    0.6913    0.6913    0.5155    0.519    0.8427    
7    0.6892    0.6892    0.6807    0.6807    0.5487    0.53    0.8388    
8    0.6471    0.6471    0.5591    0.5591    0.6931    0.768    0.8438    
9    0.5232    0.5232    0.4137    0.4137    0.7795    0.8165    0.8916    
10    0.4411    0.4411    0.4122    0.4122    0.8202    0.814    0.9018