Add new if condition.

SC-SGS · Apr 26, 2024 · ef063f1 · ef063f1
1 parent af31a61
commit ef063f1
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/include/plssvm/parameter.hpp b/include/plssvm/parameter.hpp
@@ -204,8 +204,10 @@ struct parameter {
         }
         if constexpr (parser.has(plssvm::gamma)) {
             // get the value of the provided named parameter
-            if constexpr (std::is_same_v<detail::remove_cvref_t<decltype(parser(plssvm::gamma))>, gamma_coefficient_type>) {
-                gamma = detail::get_value_from_named_parameter<gamma_coefficient_type>(parser, plssvm::gamma);
+            if constexpr (std::is_same_v<detail::remove_cvref_t<decltype(parser(plssvm::gamma))>, gamma_type>) {
+                gamma = static_cast<gamma_type>(parser(plssvm::gamma));
+            } else if constexpr (std::is_same_v<detail::remove_cvref_t<decltype(parser(plssvm::gamma))>, gamma_coefficient_type>) {
+                gamma = static_cast<gamma_coefficient_type>(parser(plssvm::gamma));
             } else {
                 gamma = detail::get_value_from_named_parameter<real_type>(parser, plssvm::gamma);
             }