Add a test set to choose the test learing rate