autoreg.py 文件源码

def eval(self, ss, returnResid=False, *args, **kwargs):
        ss = util.segmat(ss)

        preds = []
        gi = []
        for i in xrange(ss.shape[2]):
            v = ss[:,:,i]

            xs = self.getInputs(v)
            gs = self.getTargets(v)

            preds.append(self.model[i].evals(xs, *args, **kwargs).squeeze(2))

            if returnResid:
                gi.append(gs.squeeze(2))

        preds = np.rollaxis(np.array(preds), 0,3)

        if returnResid:
            gs = np.rollaxis(np.array(gi), 0,3)
            resids = gs - preds
            return preds, resids
        else:
            return preds