ensure gamma is between 0 and 1

bigartm · Mar 28, 2018 · 258df36 · 258df36
1 parent 16f9dc1
commit 258df36
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/artm/core/check_messages.h b/src/artm/core/check_messages.h
@@ -564,6 +564,12 @@ inline void FixMessage(::artm::RegularizerConfig* message) {
       BOOST_THROW_EXCEPTION(InternalError("Given RegularizerType is not supported for json serialization"));
     }
   }
+
+  if (message->has_gamma() && (message->gamma() < 0) || (message->gamma() > 1)) {
+    BOOST_THROW_EXCEPTION(InvalidOperation(
+      "Regularization parameter 'gamma' must be between 0 and 1. "
+      "Refer to documentation for more details. "));
+  }
 }
 
 template<>