From 486690f31dfe8da33fbd0711137844424d0321eb Mon Sep 17 00:00:00 2001 From: Birte Kristina Friesel Date: Mon, 12 Feb 2024 10:51:42 +0100 Subject: unfuck param_names / feature_names handling --- lib/utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'lib/utils.py') diff --git a/lib/utils.py b/lib/utils.py index 989c830..d6cdfc5 100644 --- a/lib/utils.py +++ b/lib/utils.py @@ -315,8 +315,8 @@ def param_to_ndarray( for i, paramset in distinct_values.items(): distinct_values[i] = sorted(paramset) category_to_scalar[i] = dict() - for j, param in enumerate(distinct_values[i]): - category_to_scalar[i][param] = j + for j, param_value in enumerate(distinct_values[i]): + category_to_scalar[i][param_value] = j ignore_index = dict() for i in range(len(param_tuples[0])): -- cgit v1.2.3