From c1dfcd4dca136da8dac5a2ec09e15d744f905734 Mon Sep 17 00:00:00 2001 From: Daniel Friesel Date: Wed, 15 Jun 2022 08:43:04 +0200 Subject: param_to_ndarray: handle parameters with numeric and non-numeric values --- lib/utils.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/lib/utils.py b/lib/utils.py index ed21f0f..77f9954 100644 --- a/lib/utils.py +++ b/lib/utils.py @@ -244,8 +244,10 @@ def param_to_ndarray( ret_tuple = list() for i, param in enumerate(param_tuple): if not ignore_index[i]: - if i in category_to_scalar: + if i in category_to_scalar and not is_numeric(param): ret_tuple.append(category_to_scalar[i][param]) + elif categorial_to_scalar: + ret_tuple.append(soft_cast_int(param)) else: ret_tuple.append(param) ret_tuples.append(ret_tuple) -- cgit v1.2.3