Add NMSE loss and shrinkage to tf models

[arrrobase]

• [x] add tf compatible NMSE loss function
• [x] add tf compatible shrinkage
• [x] add option of averaging over cells
• [x] add keyword for loss types