diff options
author | Daniel Friesel <daniel.friesel@uos.de> | 2020-07-09 14:55:08 +0200 |
---|---|---|
committer | Daniel Friesel <daniel.friesel@uos.de> | 2020-07-09 14:55:08 +0200 |
commit | 016210b6c64881a683b45a91af213a1d1a1128ab (patch) | |
tree | 48f700bf5c1039ca79cd7f3e69c1f48fdd4595e3 /lib/utils.py | |
parent | 2f5fa87125ccf44f1c6e208ed736070274ea6e2e (diff) |
working candidate detection for decision trees
Diffstat (limited to 'lib/utils.py')
-rw-r--r-- | lib/utils.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/lib/utils.py b/lib/utils.py index d28ecda..990ad01 100644 --- a/lib/utils.py +++ b/lib/utils.py @@ -204,6 +204,7 @@ def filter_aggregate_by_param(aggregate, parameters, parameter_filter): indices_to_keep = list( map(lambda x: x[param_index] == param_value, aggregate[name]["param"]) ) + # ["param"] is not a numpy array, so we can't use ["param"][indices_to_keep] and rely on this map-filter-expression instead aggregate[name]["param"] = list( map( lambda iv: iv[1], |