From 09ca8017e5e79a242a1bfcabdc55ba63427d8372 Mon Sep 17 00:00:00 2001 From: Jorge Fabila Date: Wed, 13 Aug 2025 15:50:34 +0200 Subject: [PATCH] change in format labels --- client_cmd.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/client_cmd.py b/client_cmd.py index 19bdd57..c74a0c3 100644 --- a/client_cmd.py +++ b/client_cmd.py @@ -53,6 +53,17 @@ args = parser.parse_args() config = vars(args) + new = [] + for i in config["train_labels"]: + parsed = i.replace("]", "").replace("[", "").replace(",", "") + new.append(parsed) + config["train_labels"] = new + + new = [] + for i in config["target_label"]: + parsed = i.replace("]", "").replace("[", "").replace(",", "") + new.append(parsed) + config["target_labels"] = new if config["model"] in ("logistic_regression", "elastic_net", "lsvc"): config["linear_models"] = {}