set the correct loss_improvement for the AverageLearner
Currently test_expected_loss_improvement_is_less_than_total_loss fails because we set an incorrect loss_improvement.
Currently test_expected_loss_improvement_is_less_than_total_loss fails because we set an incorrect loss_improvement.