def __init__(self, axis='all', eps=1e-8):
        df.Criterion.__init__(self)
        self.axis = axis
        self.eps = eps
- axis: 'all' means all except the batch-axis.
dataset/ETHPy150Open lucasb-eyer/DeepFried2/DeepFried2/criteria/RMSECriterion.py/RMSECriterion.__init__