From 8ba941bf000dde780de83ff412bbde8f6b9967fe Mon Sep 17 00:00:00 2001 From: Gene Date: Sun, 26 Nov 2023 18:04:56 +0800 Subject: [PATCH] [MNT] modify check details for classification tasks --- learnware/market/easy/checker.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/learnware/market/easy/checker.py b/learnware/market/easy/checker.py index db19ca3..7e0983e 100644 --- a/learnware/market/easy/checker.py +++ b/learnware/market/easy/checker.py @@ -47,6 +47,10 @@ class EasySemanticChecker(BaseChecker): if semantic_spec["Task"]["Values"][0] in ["Classification", "Regression"]: assert semantic_spec["Output"] is not None, "Lack of output semantics" dim = semantic_spec["Output"]["Dimension"] + assert ( + dim > 1 or semantic_spec["Task"]["Values"][0] == "Regression" + ), "Classification task must have dimension > 1" + for k, v in semantic_spec["Output"]["Description"].items(): assert int(k) >= 0 and int(k) < dim, f"Dimension number in [0, {dim})" assert isinstance(v, str), "Description must be string"