示例#1
0
 def get_score(self):
     self.outcome = [x for _, x in enumerate(self.Is)]
     self.scores = {}
     self.scores['SSE'] = RSS(self.outcome, self.CDC_obs, 1)
     self.scores['RMSE'] = RMSE(self.CDC_obs, self.outcome, 1)
     self.scores['MSPE'] = MSPE(self.CDC_obs, self.outcome, 1)
     self.scores['CORR'] = corrcoef(self.CDC_obs, self.outcome, 1)
示例#2
0
    def get_score(self):
        I_mat = as_array(self.Is)
        for i, w in enumerate(self.weights):
            I_mat[i] *= w

        self.IS = sum(I_mat, axis=1)

        time_gap = 1  # self.epochs / 52
        idx = [x for x in range(self.epochs) if not x % time_gap]

        self.score = RSS(self.CDC_obs, self.IS[idx])
        self.scores = {}
        self.scores['SSE'] = self.score
        self.scores['RMSE'] = RMSE(self.CDC_obs, self.IS[idx])
        self.scores['MSPE'] = MSPE(self.CDC_obs, self.IS[idx])
        self.scores['CORR'] = corrcoef(self.CDC_obs, self.IS[idx])[0, 1]
        return self.score