RMSE helper function for multiple outputs
Recommendation: weak & strong train / test split
Recommendation: better train / test split for weak prediction