disable `reg_sqrt` in huber loss

f660b5fe · Guolin Ke · GitHub · e5750f18 · f660b5fe
Unverified Commit f660b5fe authored May 31, 2018 by Guolin Ke Committed by GitHub May 31, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 8 additions and 1 deletion

src/objective/regression_objective.hpp src/objective/regression_objective.hpp +8 -1

No files found.
--- a/src/objective/regression_objective.hpp
+++ b/src/objective/regression_objective.hpp
@@ -262,10 +262,17 @@ class RegressionHuberLoss: public RegressionL2loss {
 public:
  explicit RegressionHuberLoss(const Config& config): RegressionL2loss(config) {
    alpha_ = static_cast<double>(config.alpha);
+    if (sqrt_) {
+      Log::Warning("Cannot use sqrt transform in %s Regression, will auto disable it", GetName());
+      sqrt_ = false;
+    }
  }

  explicit RegressionHuberLoss(const std::vector<std::string>& strs): RegressionL2loss(strs) {
-
+    if (sqrt_) {
+      Log::Warning("Cannot use sqrt transform in %s Regression, will auto disable it", GetName());
+      sqrt_ = false;
+    }
  }

  ~RegressionHuberLoss() {