param_to_ndarray: handle parameters with numeric and non-numeric values

1 files changed, 3 insertions, 1 deletions

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)