Add R2
sims1253 opened this issue · 6 comments
It would be nice if the loo objects weren't hardcoded to use elpd. For a study I wanted to include rmse and R2 and had to build fake loo objects in order to use the loo_compare functionality. Code is here.
@sims1253 do you need something else than what is in the PRs mentioned above, or can we close this issue?
We do have rmse, but we're still missing R2 (which is available in brms and rstanarm, but not as plain loo package version)
...which is available in brms
At least the brms version does not use psis, which is why I wrote the version from my pr.
brms loo_R2
does use psis based on the code https://github.com/paul-buerkner/brms/blob/d1a050803670f4f4647324ce9a84780591a40692/R/loo_predict.R#L202
True. Sorry, I'm currently on vacation and only have my phone and only found the Bayes r2 function.
Maybe it was about the pointwise estimate for the SE then that we built the new version. I don't quite remember to be honest.