From d32d5e49727161dfab34751a837ef78165da1ea8 Mon Sep 17 00:00:00 2001 From: Birte Kristina Friesel Date: Mon, 29 Jan 2024 09:01:46 +0100 Subject: LMT: Fix DFATOOL_LMT_MIN_SAMPLES_LEAF default value --- lib/parameters.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'lib') diff --git a/lib/parameters.py b/lib/parameters.py index c3af6b0..73bfca1 100644 --- a/lib/parameters.py +++ b/lib/parameters.py @@ -1190,7 +1190,9 @@ class ModelAttribute: # `ceil(min_samples_leaf * n_samples)` are the minimum # number of samples for each node. if "." in os.getenv("DFATOOL_LMT_MIN_SAMPLES_LEAF", "0.1"): - min_samples_leaf = float(os.getenv("DFATOOL_LMT_MIN_SAMPLES_LEAF")) + min_samples_leaf = float( + os.getenv("DFATOOL_LMT_MIN_SAMPLES_LEAF", "0.1") + ) else: min_samples_leaf = int(os.getenv("DFATOOL_LMT_MIN_SAMPLES_LEAF")) -- cgit v1.2.3