faster RMSE computation / experiment_example2.py