Tested with 2 hidden layers with scale out. Add MSE for testing dataset.