Add support for Double arrays.
Add mse_train and mse_test.