where i runs over all samples in the batch, N batch is the number of samples per batch, E a,i,0 refers to the reference activation energy, and E a,i,p refers to the predicted activation energy. For the multi-task model (G→ ∆H r , Ea) trained to predict both ∆H r and E a , the loss function was weighted to prioritize E a accuracy according to