epoch    tot_tloss    gw_tloss    tot_vloss    gw_vloss    train_acc    valid_acc    roc_auc    
0    0.1483    0.1483    0.0838    0.0838    0.9395    0.971    0.9953    
1    0.0792    0.0792    0.0656    0.0656    0.9745    0.9781    0.9972    
2    0.0655    0.0655    0.0553    0.0553    0.9787    0.9828    0.998    
3    0.0572    0.0572    0.0567    0.0567    0.9818    0.9836    0.998    
4    0.0539    0.0539    0.0484    0.0484    0.9826    0.9847    0.9984    
5    0.0484    0.0484    0.0427    0.0427    0.9847    0.9865    0.9986    
6    0.0467    0.0467    0.0431    0.0431    0.9853    0.986    0.9987    
7    0.0453    0.0453    0.0414    0.0414    0.9858    0.9871    0.9988    
8    0.0435    0.0435    0.0438    0.0438    0.9864    0.9868    0.9988    
9    0.0419    0.0419    0.0364    0.0364    0.9869    0.9884    0.9988    
10    0.0419    0.0419    0.0367    0.0367    0.9866    0.9886    0.999    
11    0.0408    0.0408    0.0397    0.0397    0.987    0.9881    0.9988    
12    0.0397    0.0397    0.0377    0.0377    0.9874    0.9875    0.999    
13    0.0375    0.0375    0.038    0.038    0.9881    0.9883    0.999    
14    0.0365    0.0365    0.0347    0.0347    0.9883    0.9896    0.999    
15    0.0362    0.0362    0.0369    0.0369    0.9886    0.9884    0.9989    
16    0.0374    0.0374    0.0344    0.0344    0.988    0.9892    0.9991    
17    0.0368    0.0368    0.0308    0.0308    0.9881    0.99    0.9992    
18    0.0358    0.0358    0.0358    0.0358    0.9887    0.9896    0.999    
19    0.0358    0.0358    0.033    0.033    0.9886    0.9898    0.9991    
20    0.0362    0.0362    0.0328    0.0328    0.9881    0.9898    0.9991    
21    0.0343    0.0343    0.0395    0.0395    0.9889    0.9887    0.999    
22    0.0357    0.0357    0.0334    0.0334    0.9888    0.9897    0.9992    
23    0.0333    0.0333    0.0332    0.0332    0.9895    0.9899    0.9992    
24    0.034    0.034    0.0342    0.0342    0.9893    0.9901    0.9991