diff options
| author | Birte Kristina Friesel <birte.friesel@uos.de> | 2024-01-29 09:01:46 +0100 |
|---|---|---|
| committer | Birte Kristina Friesel <birte.friesel@uos.de> | 2024-01-29 09:01:46 +0100 |
| commit | d32d5e49727161dfab34751a837ef78165da1ea8 (patch) | |
| tree | 018e464be56fd539d686b90075e4018d2c62c8bb /lib/parameters.py | |
| parent | 90e69c307c939223e50a693f69967c7a7a110f73 (diff) | |
LMT: Fix DFATOOL_LMT_MIN_SAMPLES_LEAF default value
Diffstat (limited to 'lib/parameters.py')
| -rw-r--r-- | lib/parameters.py | 4 |
1 files changed, 3 insertions, 1 deletions
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")) |