From dd12098603ac415904b9a8d512889deb995a8391 Mon Sep 17 00:00:00 2001 From: Jordan Gong Date: Thu, 7 Jan 2021 20:54:44 +0800 Subject: Train different models in different conditions --- models/model.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) (limited to 'models/model.py') diff --git a/models/model.py b/models/model.py index 1dc0f23..54f3441 100644 --- a/models/model.py +++ b/models/model.py @@ -220,13 +220,15 @@ class Model: return self._gen_sig(list(_config.values())) - def _gen_sig(self, values: Union[tuple, list, str, int, float]) -> str: + def _gen_sig(self, values: Union[tuple, list, set, str, int, float]) -> str: strings = [] for v in values: if isinstance(v, str): strings.append(v) - elif isinstance(v, (tuple, list)): + elif isinstance(v, (tuple, list, set)): strings.append(self._gen_sig(v)) + elif isinstance(v, dict): + strings.append(self._gen_sig(list(v.values()))) else: strings.append(str(v)) return '_'.join(strings) -- cgit v1.2.3