Ok, I'm also with weighing the outputs for each dimension, after computing the r2_score. I'm waiting for a final go ahead, before I can work on a commit.