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)
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