minor speedup: no need to iterate further

BayesWitnesses · Jun 24, 2020 · 2b15830 · 2b15830
1 parent 00b3294
commit 2b15830
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/m2cgen/assemblers/boosting.py b/m2cgen/assemblers/boosting.py
@@ -263,6 +263,7 @@ def _bin_class_sigmoid_transform(self, expr, to_reuse=True):
             config_entry = config_part.split(":")
             if config_entry[0] == "sigmoid":
                 coef = np.float64(config_entry[1])
+                break
         return super()._bin_class_convert_output(
             utils.mul(ast.NumVal(coef), expr), to_reuse=to_reuse)